Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 
 
-def generate(text, model, tokenizer,device,do_sample,top_k=100, epsilon_cutoff=.00005, temperature=1):
+def vipe_generate(text, model, tokenizer,device,do_sample,top_k=100, epsilon_cutoff=.00005, temperature=1):
     #mark the text with special tokens
     text=[tokenizer.eos_token +  i + tokenizer.eos_token for i in text]
     batch=tokenizer(text, padding=True, return_tensors="pt")
@@ -28,8 +28,8 @@ tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
 tokenizer.pad_token = tokenizer.eos_token
 
 
-def generate(text):
-    result = gprompts=generate(text,model,tokenizer,do_sample=True,device=device)
+def vipe_generate(text):
+    result =generate(text,model,tokenizer,do_sample=True,device=device)
     return result
 
 examples = [
@@ -39,9 +39,9 @@ examples = [
 ]
 
 demo = gr.Interface(
-    fn=generate,
-    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Generated Text"),
+    fn=vipe_generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Arbitrary Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Prompt for Visualizations"),
     examples=examples
 )