Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import base64
 from io import BytesIO
 from PIL import Image  # Corrected import
+import random 
 
 # Get API key from environment variable
 api_key = os.environ.get("NVCF_API_KEY")
@@ -21,6 +22,8 @@ headers = {
 
 # Function to generate image using the API
 def generate_image(prompt, negative_prompt, sampler, seed, guidance_scale, inference_steps):
+    if seed is None or seed == 0:
+        seed = random.randint(11111111, 999999999999999)
     payload = {
         "prompt": prompt,
         "negative_prompt": negative_prompt,
@@ -62,10 +65,10 @@ iface = gr.Interface(
     inputs=[
         gr.Textbox(label="Prompt", placeholder="Describe the image you want to generate"),
         gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image"),
-        gr.Dropdown(label="Sampler", choices=["DPM", "DDPM", "PLMS"], value="DPM"),
+        gr.Dropdown(label="Sampler", choices=["DPM", "EulerA", "LMS", "DDIM"], value="DPM"),
         gr.Number(label="Seed", value=0),
-        gr.Slider(label="Guidance Scale", minimum=0, maximum=20, value=5),
-        gr.Slider(label="Inference Steps", minimum=1, maximum=50, value=25)
+        gr.Slider(label="Guidance Scale", minimum=1, maximum=9, value=5),
+        gr.Slider(label="Inference Steps", minimum=5, maximum=100, value=35)
     ],
     outputs=gr.Image(label="Generated Image"),
 )