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)