Update app.py

app.py

@@ -23,7 +23,9 @@ def bot_streaming(message, history):
     for hist in history:
       if type(hist[0])==tuple:
         image = hist[0][0]
-  
+
+  if image is None:
+      gr.Error("You need to upload an image for LLaVA to work.")
   prompt=f"[INST] <image>\n{message['text']} [/INST]"
   image = Image.open(image).convert("RGB")
   inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")