Update app.py

app.py

@@ -34,7 +34,6 @@ image_processor = CLIPImageProcessor.from_pretrained(vision_tower_name)
 ##########################################################################
 def generate_prompt(instruction):
     instruction = instruction.strip().replace('\r\n','\n').replace('\n\n','\n')
-    input = input.strip().replace('\r\n','\n').replace('\n\n','\n')
     return f"\n{instruction}\n\nAssistant:"
 
 def generate(
@@ -104,7 +103,7 @@ examples = [
 def chatbot(image, question):
     image = image_processor(images=image.convert('RGB'), return_tensors='pt')['pixel_values']
     image_features = visual_encoder.encode_images(image.unsqueeze(0))
-    emb_mixer.set_image_embeddings(image_features)
+    emb_mixer.set_image_embeddings(image_features.squeeze(0))
     model.w["emb.weight"] = emb_mixer.get_input_embeddings()
     image_ids = [i for i in range(emb_mixer.image_start_index, emb_mixer.image_start_index + len(image_features))]
     input_text = generate_prompt(question)
@@ -116,13 +115,13 @@ with gr.Blocks(title=title) as demo:
         with gr.Column():
             image = gr.Image(type='pil', label="Image")
         with gr.Column():
-            prompt = gr.Textbox(lines=3, label="Prompt", 
-                value="Assistant: Please upload an image and ask a question.")
+            prompt = gr.Textbox(lines=5, label="Prompt", 
+                value="Please upload an image and ask a question.")
             with gr.Row():
                 submit = gr.Button("Submit", variant="primary")
                 clear = gr.Button("Clear", variant="secondary") 
         with gr.Column():
-            output = gr.Textbox(label="Output", lines=5)
+            output = gr.Textbox(label="Output", lines=7)
     data = gr.Dataset(components=[image, prompt], samples=examples, label="Examples", headers=["Image", "Prompt"])
     submit.click(chatbot, [image, prompt], [output])
     clear.click(lambda: None, [], [output])