Update app.py

app.py

@@ -9,7 +9,7 @@ def get_caption(image_in):
 	    image_in,	# str representing input in 'raw_image' Image component
 	    api_name="/predict"
     )
-    print(fusecap_result)
+    print(f"IMAGE CAPTION: {fusecap_result}")
     return fusecap_result
 
 import re
@@ -40,10 +40,11 @@ instruction = f"""
 """
 
 def infer(image_in):
-
+    gr.Info("Getting image caption from Fuse Cap...")
     user_prompt = get_caption(image_in)
     prompt = f"{instruction.strip()}\n{user_prompt}</s>"    
     print(f"PROMPT: {prompt}")
+    gr.Info("Building a system according to the image caption ...")
     outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
     print(outputs)