Update app.py

app.py

@@ -17,6 +17,7 @@ processor = AutoProcessor.from_pretrained("ahmed-masry/ChartInstruct-LLama2")
 @spaces.GPU
 def predict(image, input_text):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    model = model.to(device)
 
     input_prompt = f"<image>\n Question: {input_text} Answer: "
     image = image.convert("RGB")