Update main.py

main.py

@@ -173,7 +173,8 @@ class InputImage(BaseModel):
 
 @app.post("/Immagine")
 def generate_image(request: Request, input_data: InputImage):
-    client = Client("https://hysts-sd-xl.hf.space/--replicas/cplz0/")
+    #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)
@@ -191,24 +192,45 @@ 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
-        		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"    
-            )
+                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()