Update app.py

app.py

@@ -8,6 +8,7 @@ import logging
 import math 
 from tqdm import tqdm
 import time
+import spaces
 
 #logging.basicConfig(level=logging.DEBUG)
 
@@ -27,7 +28,7 @@ def update_selection(selected_state: gr.SelectData):
         selected_state
     )
 
-
+@spaces.GPU
 def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     logging.debug(f"Inside run_lora, selected_state: {selected_state}")
     if not selected_state: