Update app.py

app.py

@@ -1,28 +1,31 @@
 import gradio as gr
 from transformers import pipeline
+import os
+os.system("pip install -r requirements.txt")
 
-pipe = pipeline('text-generation', model_id='Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator')
-
-def get_valid_prompt(text: str) -> str:
-    dot_split = text.split('.')[0]
-    n_split = text.split('\n')[0]
+pipe = pipeline(
+    "text-generation",
+    model="Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator",
+    tokenizer="gpt2"
+)
 
-    return {
-        len(dot_split) < len(n_split): dot_split,
-        len(n_split) > len(dot_split): n_split,
-        len(n_split) == len(dot_split): dot_split   
-    }[True]
 
-def generate_prompt(prompt):
-    valid_prompt = get_valid_prompt(pipe(prompt, max_length=77)[0]['generated_text'])
-    return valid_prompt
+def generate_text(prompt):
+    return pipe(prompt, max_length=77)[0]["generated_text"]
 
 iface = gr.Interface(
-    fn=generate_prompt,
-    inputs="text",
-    outputs="text",
-    title="Prompt Generator",
-    description="Enter a prompt and get the valid prompt generated by the script."
+    fn=generate_text,
+    
+    #input is a text box
+    inputs=gr.Textbox(lines=5, label="Prompt"),
+    
+    
+    # output is a text box with copy button
+    outputs=gr.Textbox(label="Output", show_copy_button=True),
+    
+    title="GPT-2 650k Stable Diffusion Prompt Generator",
+    description="GPT-2 650k Stable Diffusion Prompt Generator",
+    api_name="predict"
 )
 
-iface.launch()
+iface.launch(show_api=True)