Update main.py

main.py

@@ -167,14 +167,14 @@ class InputImage(BaseModel):
     input: str
     negativePrompt: str = '' 
     style: str = ''
-    steps: int = 30
-    cfg: int = 7
+    steps: int = 25
+    cfg: int = 6
     seed: int = -1
 
 @app.post("/Immagine")
 def generate_image(request: Request, input_data: InputImage):
-    #client = Client("https://hysts-sd-xl.hf.space/--replicas/cplz0/")
-    client = Client("https://segmind-segmind-stable-diffusion.hf.space/--replicas/xc5bq/")
+    client = Client("https://hysts-sd-xl.hf.space/--replicas/cplz0/")
+    #client = Client("https://segmind-segmind-stable-diffusion.hf.space/--replicas/xc5bq/")
     
     if input_data.style:    
         print(input_data.style)
@@ -192,44 +192,44 @@ def generate_image(request: Request, input_data: InputImage):
     attempt = 0
     while attempt < max_attempts:
         try:
-            #result = client.predict(
-            #    input_data.input,	# str  in 'Prompt' Textbox component
-        	#	input_data.negativePrompt,	# str  in 'Negative prompt' Textbox component
-        	#	input_data.input,	# str  in 'Prompt 2' Textbox component
-        	#	input_data.negativePrompt,	# str  in 'Negative prompt 2' Textbox component
-        	#	True,	# bool  in 'Use negative prompt' Checkbox component
-        	#	True,	# bool  in 'Use prompt 2' Checkbox component
-        	#	True,	# bool  in 'Use negative prompt 2' Checkbox component
-        	#	input_data.seed,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
-        	#	1024,	# float (numeric value between 256 and 1024) in 'Width' Slider component
-        	#	1024,	# float (numeric value between 256 and 1024) in 'Height' Slider component
-        	#	input_data.cfg,	# float (numeric value between 1 and 20) in 'Guidance scale for base' Slider component
-        	#	input_data.cfg,	# float (numeric value between 1 and 20) in 'Guidance scale for refiner' Slider component
-        	#	input_data.steps,	# float (numeric value between 10 and 100) in 'Number of inference steps for base' Slider component
-        	#	input_data.steps,	# float (numeric value between 10 and 100) in 'Number of inference steps for refiner' Slider component
-        	#	True,	# bool  in 'Apply refiner' Checkbox component
-        	#	api_name="/run"    
-            #)
             result = client.predict(
-                input_data.input, # str  in 'Prompt' Textbox component
-                input_data.negativePrompt,	# str  in 'Negative prompt' Textbox component
-                "(No style)",	# str  in 'Image Style' Radio component
-                "", # str  in 'Prompt 2' Textbox component
-                "",	# str  in 'Negative prompt 2' Textbox component
-                True,	# bool  in 'Use negative prompt' Checkbox component
-                False,	# bool  in 'Use prompt 2' Checkbox component
-                False,	# bool  in 'Use negative prompt 2' Checkbox component
-                0,	# int | float (numeric value between 0 and 2147483647) in 'Seed' Slider component
-                1024,	# int | float (numeric value between 256 and 1024) in 'Width' Slider component
-                1024,	# int | float (numeric value between 256 and 1024) in 'Height' Slider component
-                input_data.cfg,	# int | float (numeric value between 1 and 20) in 'Guidance scale for base' Slider component
-                input_data.cfg,	# int | float (numeric value between 1 and 20) in 'Guidance scale for refiner' Slider component
-                input_data.steps,	# int | float (numeric value between 10 and 100) in 'Number of inference steps for base' Slider component
-                input_data.steps,	# int | float (numeric value between 10 and 100) in 'Number of inference steps for refiner' Slider component
-                True,	# bool  in 'Apply refiner' Checkbox component
-                True,	# bool  in 'Randomize seed' Checkbox component
-                api_name="/run"
+                input_data.input,	# str  in 'Prompt' Textbox component
+        		input_data.negativePrompt,	# str  in 'Negative prompt' Textbox component
+        		input_data.input,	# str  in 'Prompt 2' Textbox component
+        		input_data.negativePrompt,	# str  in 'Negative prompt 2' Textbox component
+        		True,	# bool  in 'Use negative prompt' Checkbox component
+        		True,	# bool  in 'Use prompt 2' Checkbox component
+        		True,	# bool  in 'Use negative prompt 2' Checkbox component
+        		input_data.seed,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
+        		1024,	# float (numeric value between 256 and 1024) in 'Width' Slider component
+        		1024,	# float (numeric value between 256 and 1024) in 'Height' Slider component
+        		input_data.cfg,	# float (numeric value between 1 and 20) in 'Guidance scale for base' Slider component
+        		input_data.cfg,	# float (numeric value between 1 and 20) in 'Guidance scale for refiner' Slider component
+        		input_data.steps,	# float (numeric value between 10 and 100) in 'Number of inference steps for base' Slider component
+        		input_data.steps,	# float (numeric value between 10 and 100) in 'Number of inference steps for refiner' Slider component
+        		True,	# bool  in 'Apply refiner' Checkbox component
+        		api_name="/run"    
             )
+            #result = client.predict(
+            #    input_data.input, # str  in 'Prompt' Textbox component
+            #    input_data.negativePrompt,	# str  in 'Negative prompt' Textbox component
+            #    "(No style)",	# str  in 'Image Style' Radio component
+            #    "", # str  in 'Prompt 2' Textbox component
+            #    "",	# str  in 'Negative prompt 2' Textbox component
+            #    True,	# bool  in 'Use negative prompt' Checkbox component
+            #    False,	# bool  in 'Use prompt 2' Checkbox component
+            #    False,	# bool  in 'Use negative prompt 2' Checkbox component
+            #    0,	# int | float (numeric value between 0 and 2147483647) in 'Seed' Slider component
+            #    1024,	# int | float (numeric value between 256 and 1024) in 'Width' Slider component
+            #    1024,	# int | float (numeric value between 256 and 1024) in 'Height' Slider component
+            #    input_data.cfg,	# int | float (numeric value between 1 and 20) in 'Guidance scale for base' Slider component
+            #    input_data.cfg,	# int | float (numeric value between 1 and 20) in 'Guidance scale for refiner' Slider component
+            #    input_data.steps,	# int | float (numeric value between 10 and 100) in 'Number of inference steps for base' Slider component
+            #    input_data.steps,	# int | float (numeric value between 10 and 100) in 'Number of inference steps for refiner' Slider component
+            #    True,	# bool  in 'Apply refiner' Checkbox component
+            #    True,	# bool  in 'Randomize seed' Checkbox component
+            #    api_name="/run"
+            #)
             image_url = result
             with open(image_url, 'rb') as img_file:
                 img_binary = img_file.read()