fix prompts

app.py

@@ -263,7 +263,7 @@ def inference(img, text, is_translation, prompt=None):
         with torch.no_grad():
             prefix = clip_model.encode_image(image).to(device, dtype=torch.float32)
             prefix_embed = model.clip_project(prefix).reshape(1, prefix_length, -1)
-            generated_text_prefix = generate_beam(model, tokenizer, embed=prefix_embed, prompt="Một bức ảnh về")[0]
+            generated_text_prefix = generate_beam(model, tokenizer, embed=prefix_embed, prompt=prompt)[0]
 
     return generated_text_prefix
   
@@ -274,7 +274,7 @@ examples=[
     ["examples/harry.jpeg","", False, "Một bức ảnh về"], 
     ["examples/OldTrafford.jpeg","", False, "Một bức ảnh về"], 
     ["examples/coffee.jpg","", False, "Một bức ảnh về"], 
-    ["", "What is your name?", True]
+    ["", "What is your name?", True, ""]
 ]
 
 inputs = [