Update app.py

app.py

@@ -1,7 +1,6 @@
-+import spaces
-
+import spaces
 from transformers import pipeline
-
+import gradio as gr
 
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')
 
@@ -22,14 +21,20 @@ 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)
 
-+@spaces.GPU
+@spaces.GPU
 def run():
     with torch.autocast('cuda', dtype=torch.bfloat16):
-        print(
+        return(
             pipe('Here is a recipe for vegan banana bread:\n',
                 max_new_tokens=100,
                 do_sample=True,
                 use_cache=True))
-run()
 
 
+
+with gr.Blocks() as app:
+    btn = gr.Button()
+    outp=gr.Textbox()
+    btn.click(run,None,outp)
+app.launch()
+