gpt2prompt

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import os 
 import sys
 from pathlib import Path
-       
+from transformers import pipeline
+pipe = pipeline('text-generation', model_id='Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator')
 models = [
     "Yntec/NovelAIRemix",
     "Joeythemonster/anything-midjourney-v-4-1",
@@ -95,8 +96,16 @@ with gr.Blocks(css=css) as myface:
         with gr.Column(scale=50):
             input_text=gr.Textbox(label="Use this box to extend an idea automatically, by typing some words and clicking Extend Idea",lines=2)
             see_prompts=gr.Button("Extend Idea")
-                
-     
+
+    def get_valid_prompt(text: str) -> str:
+      dot_split = text.split('.')[0]
+      n_split = text.split('\n')[0]
+
+      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 short_prompt(inputs):
         return(inputs)
     
@@ -105,7 +114,7 @@ with gr.Blocks(css=css) as myface:
     run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
     
     
-    see_prompts.click(text_it1,inputs=[input_text],outputs=magic1)
+    seeprompts.click(magic1 = get_valid_prompt(pipe(input_text, max_length=77)[0]['generated_text']))
     
 myface.queue(concurrency_count=200)
 myface.launch(inline=True, show_api=False, max_threads=400)