fix

app.py

@@ -32,7 +32,7 @@ from PIL import Image
 from io import BytesIO
 import re
 
-from .conversation import Chat, conv_llava_v1
+from llava.chat import Chat, conv_llava_v1
 
 # imports modules for registration
 
@@ -116,7 +116,7 @@ def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature, num_vis
 title = """<h1 align="center">Demo of MQT-LLaVA</h1>"""
 description = """<h3>This is the demo of MQT-LLaVA. Upload your images and start chatting!. <br> To use 
             example questions, click example image, hit upload, and press enter in the chatbox.</h3>"""
-article = """<p><a href='https://github.com/gordonhu608/MQT-LLaVA'><img src='https://img.shields.io/badge/Github-Code-blue'></a></p><p><a href='https://arxiv.org/abs/'><img src='https://img.shields.io/badge/Paper-ArXiv-red'></a></p>
+article = """<p><a href='https://gordonhu608.github.io/mqt-llava/'><img src='https://img.shields.io/badge/Project-Page-Green'></a></p><p><a href='https://github.com/gordonhu608/MQT-LLaVA'><img src='https://img.shields.io/badge/Github-Code-blue'></a></p><p><a href='https://arxiv.org/abs/'><img src='https://img.shields.io/badge/Paper-ArXiv-red'></a></p>
 """
 
 #TODO show examples below

conversation.py → llava/chat.py

@@ -488,7 +488,7 @@ class Chat:
         return output_text, ''  
 
     def upload_img(self, image, conv, img_list):
-        images = load_images(image)
+        images = load_images([image])
         # if isinstance(image, str):  # is a image path
         #     raw_image = Image.open(image).convert('RGB')
         #     image = self.vis_processor(raw_image).unsqueeze(0).to(self.device)