Update app.py

app.py

@@ -27,6 +27,10 @@ model.to(device)
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
 tokenizer.pad_token = tokenizer.eos_token
 
+def generate(text):
+    result =generate(text,model,tokenizer,do_sample=True,device=device)
+    return result
+
 
 examples = [
     ["Is string theory right?"],
@@ -35,7 +39,7 @@ examples = [
 ]
 
 demo = gr.Interface(
-    fn =vipe_generate(text,model,tokenizer,do_sample=True,device=device),
+    fn =generate,
     inputs=gr.inputs.Textbox(lines=5, label="Arbitrary Input Text"),
     outputs=gr.outputs.Textbox(label="Generated Prompt for Visualizations"),
     examples=examples