Update app.py

app.py

@@ -32,8 +32,10 @@ def image_vision(image_input_path, prompt):
         'tokenizer': tokenizer,
     }
     return_dict = model.predict_forward(**input_dict)
+    print(return_dict)
     answer = return_dict["prediction"] # the text format answer
-    print(answer)
+    
+    return answer
 
 def main_infer(image_input_path, prompt):