jkorstad commited on
Commit
cda9cd5
·
verified ·
1 Parent(s): 9528caf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def predict(image, text):
33
  inputs = processor(image, input_text, return_tensors="pt").to(model.device)
34
  outputs = model.generate(**inputs, max_new_tokens=250)
35
  response = processor.decode(outputs[0], skip_special_tokens=True)
36
- return response
37
 
38
  # Gradio
39
  interface = gr.Interface(
 
33
  inputs = processor(image, input_text, return_tensors="pt").to(model.device)
34
  outputs = model.generate(**inputs, max_new_tokens=250)
35
  response = processor.decode(outputs[0], skip_special_tokens=True)
36
+ return f"\n{response}"
37
 
38
  # Gradio
39
  interface = gr.Interface(