Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
 import requests
 import time
+import random
 import json
 import base64
 import os
+from transformers import pipeline, set_seed
 from io import BytesIO
 import io
 import html
@@ -433,6 +435,29 @@ with gr.Blocks(css=css) as demo:
 
             hf_text_button.click(hf_inference, inputs=[hf_prompt, hf_negative_prompt, hf_model, hf_steps, sampler, hf_cfg_scale, hf_width, hf_height,
                                                hf_seed], outputs=hf_image_output, concurrency_limit=64)
+        with gr.Tab("Prompt Generator"):
+            gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
+            with open("ideas.txt", "r") as f:
+                line = f.readlines()
+            def generate(starting_text):
+                seed = random.randint(100, 1000000)
+                set_seed(seed)
+                if starting_text == "":
+                    starting_text: str = line[random.randrange(0, len(line))].replace("\n", "").lower().capitalize()
+                    starting_text: str = re.sub(r"[,:\-–.!;?_]", '', starting_text)
+                response = gpt2_pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=4)
+                response_list = []
+                for x in response:
+                    resp = x['generated_text'].strip()
+                    if resp != starting_text and len(resp) > (len(starting_text) + 4) and resp.endswith((":", "-", "—")) is False:
+                        response_list.append(resp+'\n')
+                response_end = "\n".join(response_list)
+                response_end = re.sub('[^ ]+\.[^ ]+','', response_end)
+                response_end = response_end.replace("<", "").replace(">", "")
+
+    if response_end != "":
+        return response_end
+
         with gr.Tab("BLIP"):
             with gr.Tab("Base"):
                 gr.load("models/Salesforce/blip-image-captioning-base", title="BLIP-base")
@@ -444,7 +469,6 @@ with gr.Blocks(css=css) as demo:
         #    gr.load("models/mattmdjaga/segformer_b2_clothes", title="SegFormer Segmentation")
         with gr.Tab("Visual Question Answering"):
             gr.load("models/dandelin/vilt-b32-finetuned-vqa", title="ViLT VQA")
-        with gr.Tab("Object Detection"):
-            gr.load("models/facebook/detr-resnet-50", title="DETR OD")
+        
             
 demo.queue(max_size=80, api_open=False).launch(max_threads=256, show_api=False)