Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import TextIteratorStreamer
 import time
 from threading import Thread
 import torch
+import spaces
 
 model_id = "microsoft/Phi-3-vision-128k-instruct" 
 model = AutoModelForCausalLM.from_pretrained(model_id, device_map="cuda", trust_remote_code=True, torch_dtype="auto")
@@ -20,7 +21,7 @@ PLACEHOLDER = """
 </div>
 """
 
-#@spaces.GPU
+@spaces.GPU
 def bot_streaming(message, history):
     print(f'message is - {message}')
     print(f'history is - {history}')