Update app.py

app.py

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