Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from PIL import Image 
+import io
 import argparse
 import os
 import random
@@ -53,10 +54,15 @@ chat = Chat(model, vis_processor, device='cuda:{}'.format(0))
 print('Initialization Finished')
 
 
-def respond(img, question):
-    img.save("./img.jpg")
+def respond(imgs, question):
 
-    raw_img_list= ["./img.jpg"]
+    raw_img_list = []
+    
+    for i, img in enumerate(imgs):
+        temp_img = Image.open(io.BytesIO(img))
+        save_path = "./img" + str(i) + ".jpg"
+        temp_img.save(save_path)
+        raw_img_list.append(save_path)
     
     context = question
 
@@ -69,5 +75,6 @@ def respond(img, question):
 
     return llm_message
 
-iface = gr.Interface(fn=respond, title="Cheetah", inputs=[gr.Image(type="pil"), "text"], outputs="text")
+#iface = gr.Interface(fn=respond, title="Cheetah", inputs=[gr.Image(type="pil"), "text"], outputs="text")
+iface = gr.Interface(fn=respond, title="Cheetah", inputs=[gr.File(file_count='multiple',file_types=['image'],type='binary'), "text"], outputs="text")
 iface.launch()