Update app.py

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
 if torch.cuda.is_available():
@@ -193,7 +194,7 @@ def display_tree(start_sentence, scores, sequences, beam_indices):
     """
     return display
 
-
+@spaces.GPU
 def get_tables(input_text, number_steps, number_beams):
     inputs = tokenizer([input_text], return_tensors="pt")