Update app.py

app.py

@@ -6,15 +6,15 @@ from io import BytesIO
 import os
 import random
 
-def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces, seed, cfg, token):
+def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces, seed, cfg):
     restore_faces = bool(restore_faces)
     print(f"restore_faces: {restore_faces}, type: {type(restore_faces)}")
     # Define the API endpoint
-    apiUrl = "https://apiv2.makeai.run/v2/txt2img"
+    apiUrl = os.getenv("API_URL")
     # Define the request headers
     headers = {
         "Content-Type": "application/json",
-        "token": "514f7ecde6a5434dbab5c6579311ad82"
+        "token": os.getenv("API_TOKEN")
     }
 
     # Define the request body
@@ -64,7 +64,6 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         "controlnet_threshold_b": 200,
     }
 
-
     # Send the request
     response = requests.post(apiUrl, headers=headers, data=json.dumps(body), verify=False)
     # Print the response body if the status code is not 200
@@ -79,8 +78,7 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
        if 'results' in response_json and isinstance(response_json['results'], list) and len(response_json['results']) > 0:
            image_url = response_json['results'][0]
 
-           # Get the image from the URL
-           image_response = requests.get(image_url)
+           # Get the image from the URL           image_response = requests.get(image_url)
            image = PIL.Image.open(BytesIO(image_response.content))
 
            return image
@@ -115,13 +113,13 @@ iface = gr.Interface(
             "DPM++ SDE Karras",
             "DDIM",
             "PLMS"
-        ], label="Scheduler", default="Euler a"),
+        ], label="Scheduler", default="DPM++ SDE Karras"),
         gr.components.Slider(minimum=10, maximum=100, step=1.0,default=30, label="Steps"),
         gr.components.Slider(minimum=512, maximum=1600, default=768, label="Width"),
         gr.components.Slider(minimum=512, maximum=1600, default=768, label="Height"),
         gr.components.Slider(minimum=4, maximum=12, step=0.5, default=7.0, label="CFG"),
         gr.components.Number(label="Seed", default=None),
-        gr.inputs.Checkbox(label="Restore Faces"),
+        gr.inputs.Checkbox(label="Restore Faces", default=False),
     ], 
     outputs=gr.components.Image(),
     title="Freedom Demonstration",
@@ -143,4 +141,4 @@ Thanks to [Redmond.ai](LINK_TO_REDMOND.AI) for providing GPU Time and sponsoring
 )
 
 # Launch the app
-iface.launch(debug=True)
+iface.launch(debug=True)