Update app.py

app.py

@@ -17,7 +17,6 @@ model.to(device)
     
 
 def generate(
-    system_prompt,
     prompt,
     max_new_tokens,
     repetition_penalty,
@@ -26,8 +25,7 @@ def generate(
     top_k,
     seed
 ):
-    
-    input_text = f"{system_prompt}, {prompt}"
+    input_text = f"Expand the following prompt to add more detail: {prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
     if seed == 0:
@@ -53,8 +51,6 @@ def generate(
 
 prompt = gr.Textbox(label="Prompt", interactive=True)
 
-system_prompt = gr.Textbox(label="System Prompt", interactive=True)
-
 max_new_tokens = gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens", info="The maximum numbers of new tokens, controls how long is the output")
     
 repetition_penalty = gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty", info="Penalize repeated tokens, making the AI repeat less itself")
@@ -70,7 +66,6 @@ seed = gr.Number(value=42, interactive=True, label="Seed", info="A starting poin
 examples = [
     [
         "A storefront with 'Text to Image' written on it.",
-        "Expand the following prompt to add more detail:",
         512,
         1.2,
         0.5,
@@ -82,10 +77,10 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
+    inputs=[prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed],
     outputs=gr.Textbox(label="Better Prompt"),
     title="SuperPrompt-v1",
-    description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",
+    description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088) <br> Task Prefix: 'Expand the following prompt to add more detail:' is already setted!",
     examples=examples,
     concurrency_limit=20,
 ).launch(show_api=False)