Update modeling_gpt2vision.py

modeling_gpt2vision.py

@@ -98,7 +98,7 @@ class GPT2Vision(PreTrainedModel):
 
         
     def generate(self, question, image, max_new_tokens=30, **kwargs):
-        prompt = f"<image>\n\nQuestion: {question}\n\nAnswer:"
+        prompt = f"<image>\n\nQuestion: {question}\nAnswer:"
         print("prompt",prompt)
         batch = {"image": [image], "text": prompt}
         encoded_batch = self.tokenize_encode(batch, self.device)