Update app.py

app.py

@@ -4,8 +4,6 @@ import json
 import io
 import random
 import os
-import torch
-from transformers import pipeline
 from PIL import Image
 
 API_BASE_URL = "https://api-inference.huggingface.co/models/"
@@ -30,24 +28,10 @@ MODEL_LIST = [
 API_TOKEN = os.getenv("HF_READ_TOKEN")  # Make sure to set your Hugging Face token
 HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
-pipe = pipeline("text-generation", model="isek-ai/SDPrompt-RetNet-300M", trust_remote_code=True)
-
 def select_model(model_name):
     if model_name in MODEL_LIST:
         return f"{API_BASE_URL}{model_name}"
 
-def extend_prompt(input_text):
-    if not input_text.strip():
-        gr.Warning("Input text is empty!")
-        return None
-        
-    seed = random.randint(1, 1000000)
-    torch.manual_seed(seed)
-    
-    output = pipe(input_text, max_length=(len(input_text) + random.randint(60, 90)), num_return_sequences=4)
-
-    return output
-
 def generate_image(prompt, selected_model, is_negative=False, steps=1, cfg_scale=6, seed=None):
     if not prompt.strip():
         raise gr.Error("Cannot generate image: Input text is empty!")
@@ -94,12 +78,10 @@ with gr.Blocks(theme="soft") as playground:
             text_prompt = gr.Textbox(label="Prompt", placeholder="a cute cat", lines=1, elem_id="prompt-text-input")
             model_dropdown = gr.Dropdown(label="Model", choices=MODEL_LIST, elem_id="model-dropdown", value="runwayml/stable-diffusion-v1-5")
             gen_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
-            extend_button = gr.Button("Extend Prompt", variant='primary', elem_id="extend-button")
 
     with gr.Accordion("Advanced settings", open=False):
         negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
 
     gen_button.click(generate_image, inputs=[text_prompt, model_dropdown, negative_prompt], outputs=image_output)
-    extend_button.click(extend_prompt, inputs=text_prompt, outputs=text_prompt)
-
+    
 playground.launch(show_api=False)