Update chatbot.py

chatbot.py

@@ -202,7 +202,7 @@ def llava(user_prompt, chat_history):
     
     final_prompt = f"{system_llava}\n{prompt}"
         
-    inputs = processor(final_prompt, image, return_tensors="pt").to("cuda", torch.float16)
+    inputs = processor(final_prompt, image, return_tensors="pt").to("cpu", torch.float16)
 
     return inputs