Update app.py

app.py

@@ -10,6 +10,7 @@ from datetime import datetime, timedelta
 from threading import Thread
 
 import gradio as gr
+import spaces
 import numpy as np
 from PIL import Image
 from huggingface_hub import hf_hub_download
@@ -94,7 +95,7 @@ llm = LLM(model=MODEL_ID, tokenizer_mode="mistral")
 # -----------------------------------------------------------------------------
 # Main Generation Function
 # -----------------------------------------------------------------------------
-
+@spaces.GPU
 def generate(
     input_dict: dict,
     chat_history: list,