update

app.py

@@ -27,22 +27,22 @@ model.to("cpu")
 
 
 def llava(message, history):
- if hasattr(message, 'files') and message.files:
- image = message.files[0] 
+ if message["files"]:
+ image = message["files"][0] 
  else:
  for hist in history:
- if isinstance(hist[0], tuple):
+ if type(hist[0])==tuple:
  image = hist[0][0]
  
- txt = message.text # Assuming 'text' is an attribute
+ txt = message["text"]
+ 
  gr.Info("Analyzing image")
  image = Image.open(image).convert("RGB")
- prompt = f"user <image>\n{txt}assistant"
+ prompt = f"<|im_start|>user <image>\n{txt}<|im_end|><|im_start|>assistant"
  
  inputs = processor(prompt, image, return_tensors="pt")
  return inputs
 
-
 def extract_text_from_webpage(html_content):
  soup = BeautifulSoup(html_content, 'html.parser')
  for tag in soup(["script", "style", "header", "footer"]):