abc

app.py

@@ -5,6 +5,13 @@ import io
 import base64
 
 
+def base64_to_pil_image(base64_image: str) -> Image.Image:
+    image_stream = io.BytesIO(base64.b64decode(base64_image))
+    image = Image.open(image_stream)
+
+    return image
+
+
 def generate_image(
     prompt, seed, width, height, num_inference_steps, model_name, guidance_scale, key
 ):
@@ -21,11 +28,11 @@ def generate_image(
         },
     }
     response = requests.post(
-        "http://proxy_client_nicheimage.nichetensor.com:10003//generate", json=data
+        "http://proxy_client_nicheimage.nichetensor.com:10003/generate", json=data
     )
     base64_image = response.json()
-    image_stream = io.BytesIO(base64.b64decode(base64_image))
-    image = Image.open(image_stream)
+    print(base64_image)
+    image = base64_to_pil_image(base64_image)
     return image
 
 
@@ -34,11 +41,17 @@ iface = gr.Interface(
     inputs=[
         "text",
         gr.Number(value=0),
-        gr.Number(value=512),
-        gr.Number(value=512),
-        gr.Number(value=30),
+        gr.Slider(label="Width", minimum=0, maximum=2048, value=512, step=64),
+        gr.Slider(label="Height", minimum=0, maximum=2048, value=512, step=64),
+        gr.Slider(label="Inference Steps", minimum=0, maximum=50, value=30, step=1),
         gr.Dropdown(choices=["RealisticVision", "SDXLTurbo"], value="RealisticVision"),
-        gr.Slider(minimum=0, maximum=1, value=7),
+        gr.Slider(
+            label="Guidance Scale",
+            minimum=0,
+            maximum=1,
+            value=7,
+            step=0.1,
+        ),
         "text",
     ],
     outputs="image",
@@ -46,4 +59,4 @@ iface = gr.Interface(
     description="Enter a prompt to generate an image.",
 )
 
-iface.queue().launch()
+iface.queue().launch(share=False)