updates code

README.md

@@ -1,13 +1,12 @@
 ---
-title: Txt2img
-emoji: 📊
-colorFrom: pink
-colorTo: red
+title: lunarnaut - generación de imágenes
+emoji: 🌖
+colorFrom: black
+colorTo: black
 sdk: gradio
 sdk_version: 3.35.2
 app_file: app.py
-pinned: false
+pinned: true
 license: unknown
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -6,46 +6,34 @@ def generate(prompt, n_prompt, modelName):
 def create_demo():
     with gr.Blocks() as demo:
         with gr.Column():
-            prompt = gr.Textbox(label='Prompt')
-            n_prompt = gr.Textbox( label='Negative Prompt', value= 'low res')
-            modelName = gr.Dropdown(choices = list(models.keys()), label = "Model", value=list(models.keys())[0])
-
-            run_button = gr.Button('Generar')
-
-        gr.Markdown("lunarnaut txt2img demo")
-        result = gr.Gallery(label='Output', show_label=False, elem_id='gallery').style(columns=1, rows=1, preview=True)
-
-        inputs = [
-            prompt,
-            n_prompt,
-            modelName,
-        ]
-
-        prompt.submit(
-            fn=generate,
-            inputs=inputs,
-            outputs=result
-        )
-        n_prompt.submit(
-            fn=generate,
-            inputs=inputs,
-            outputs=result
-        )
-
-        run_button.click(
-            fn=generate,
-            inputs=inputs,
-            outputs=result
-        )
+            model_name = gr.Dropdown(choices = list(models.keys()), label = 'Modelo', value=list(models.keys())[0])
+            prompt = gr.Textbox(label='Prompt', value='A crescent moon')
+            negative_prompt = gr.Textbox( label='Negative Prompt', value='bad quality, low resolution')
+            generate_button = gr.Button('Generar')
+
+        gr.Markdown('''
+        #lunarnaut - intro a stable diffusion 
+        ---
+        1. Selecciona un Modelo
+        2. Agrega un prompt y prompt negativo
+        3. Haz clic en 'Generar'
+        ''')
+        result = gr.Gallery(label='Resultado', show_label=False, elem_id='gallery').style(columns=1, rows=1, preview=True)
+
+        inputs = [ prompt, negative_prompt, model_name ]
+
+        prompt.submit(fn=generate, inputs=inputs, outputs=result)
+        negative_prompt.submit(fn=generate, inputs=inputs, outputs=result)
+        generate_button.click(fn=generate, inputs=inputs, outputs=result)
     return demo
 
 
 if __name__ == '__main__':
     from model import Model
     models = {
-        "Stable Diffusion v1.5": Model("runwayml/stable-diffusion-v1-5"),
-        "Anything v3.0": Model("Linaqruf/anything-v3.0"),
-        "Realistic Vision v2.0": Model("SG161222/Realistic_Vision_V2.0"),
-              }
+        'Stable Diffusion v1.5': Model('runwayml/stable-diffusion-v1-5'),
+        'Anything v3.0': Model('Linaqruf/anything-v3.0'),
+        'Realistic Vision v2.0': Model('SG161222/Realistic_Vision_V2.0'),
+    }
     demo = create_demo()
     demo.queue().launch()