Update app.py

app.py

@@ -5,11 +5,11 @@ from PIL import Image
 from transformers import AutoModel, AutoTokenizer
 import spaces
 
-device = "cuda"
+device="cuda"
 
 # Load the model and tokenizer
 model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
-model = model.to(device=device)
+model = model.to(device='cuda')
 tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True)
 model.eval()
 
@@ -51,5 +51,5 @@ with gr.Blocks(theme='abidlabs/dracula_revamped') as demo:
     image_input.change(generate_response, inputs=[image_input, question_input], outputs=output_text)
     gr.HTML(footer)
 
-# Launch the app with a longer timeout
-spaces.launch(demo, debug=True, timeout=600)
+# Launch the app
+demo.launch(debug=True)