Update app.py

app.py

@@ -62,7 +62,7 @@ def generate(
             out, state = model.forward(embs=input_embs, state=None)
         else:
             input_ids = [token]
-            out, state = model.forward(input_ids, state)
+            out, state = model.forward(tokens=input_ids, state=state)
         for n in occurrence:
             out[n] -= (args.alpha_presence + occurrence[n] * args.alpha_frequency)
 
@@ -102,6 +102,9 @@ examples = [
     ]
 ]
 def chatbot(image, question):
+    if image is None:
+        yield "Please upload an image."
+        return
     image = image_processor(images=image.convert('RGB'), return_tensors='pt')['pixel_values']
     image_features = visual_encoder.encode_images(image.unsqueeze(0)).squeeze(0) # [L, D]
     input_text = generate_prompt(question)