Update app.py

app.py

@@ -1,3 +1,5 @@
++import spaces
+
 from transformers import pipeline
 
 
@@ -20,14 +22,14 @@ PROMPT_FOR_GENERATION_FORMAT = """{intro}
 example = "James decides to run 3 sprints 3 times a week. He runs 60 meters each sprint. How many total meters does he run a week? Explain before answering."
 fmt_ex = PROMPT_FOR_GENERATION_FORMAT.format(instruction=example)
 
-
-
-with torch.autocast('cuda', dtype=torch.bfloat16):
-    print(
-        pipe('Here is a recipe for vegan banana bread:\n',
-            max_new_tokens=100,
-            do_sample=True,
-            use_cache=True))
-
++@spaces.GPU
+def run():
+    with torch.autocast('cuda', dtype=torch.bfloat16):
+        print(
+            pipe('Here is a recipe for vegan banana bread:\n',
+                max_new_tokens=100,
+                do_sample=True,
+                use_cache=True))
+run()