Update app.py

app.py

@@ -31,9 +31,7 @@ model = AutoModelForCausalLM.from_pretrained(
 
 
 
-text_only_template = """A chat between a curious user and an artificial intelligence assistant. \
-The assistant gives helpful, detailed, and polite answers to the user's questions. \
-USER: {} ASSISTANT:"""
+text_only_template = """USER: {} ASSISTANT:"""
 
 
 @spaces.GPU
@@ -44,7 +42,7 @@ def generate_caption(image, prompt):
     
     # image = Image.open(image_path).convert('RGB')
     image = image.convert('RGB')
-    query = text_only_template.format(query)
+    query = "USER: %s ASSISTANT:" % prompt
     input_by_model = model.build_conversation_input_ids(
         tokenizer,
         query=query,