Update main.py

main.py

@@ -429,8 +429,8 @@ class InputImage(BaseModel):
 
 @app.post("/Immagine")
 def generate_image(request: Request, input_data: InputImage):
-    client = Client("https://manjushri-sdxl-1-0.hf.space/")
-    
+    #client = Client("https://manjushri-sdxl-1-0.hf.space/")
+    client = Client("AP123/SDXL-Lightning")
     if input_data.style:    
         print(input_data.style)
         if input_data.style == 'RANDOM':
@@ -442,26 +442,16 @@ def generate_image(request: Request, input_data: InputImage):
             style_info = style_image[input_data.style]
             input_data.input = style_info["descrizione"].format(prompt=input_data.input)
             input_data.negativePrompt = style_info["negativePrompt"]
-    max_attempts = 2
+    max_attempts = 5
     attempt = 0
     while attempt < max_attempts:
         try:
             result = client.predict(
-               input_data.input,	# str  in 'What you want the AI to generate. 77 Token Limit. A Token is Any Word, Number, Symbol, or Punctuation. Everything Over 77 Will Be Truncated!' Textbox component
-               input_data.negativePrompt,	# str  in 'What you Do Not want the AI to generate. 77 Token Limit' Textbox component
-               1024,	# int | float (numeric value between 512 and 1024) in 'Height' Slider component
-               1024,	# int | float (numeric value between 512 and 1024) in 'Width' Slider component
-               input_data.cfg,	# int | float (numeric value between 1 and 15) in 'Guidance Scale: How Closely the AI follows the Prompt' Slider component
-               input_data.steps,	# int | float (numeric value between 25 and 100) in 'Number of Iterations' Slider component
-               0,	# int | float (numeric value between 0 and 999999999999999999) in 'Seed: 0 is Random' Slider component
-               "Yes",	# str  in 'Upscale?' Radio component
-               "",	# str  in 'Embedded Prompt' Textbox component
-               "",	# str  in 'Embedded Negative Prompt' Textbox component
-               0.99,	# int | float (numeric value between 0.7 and 0.99) in 'Refiner Denoise Start %' Slider component
-               100,	# int | float (numeric value between 1 and 100) in 'Refiner Number of Iterations %' Slider component
-               api_name="/predict"
-            ) 
-            image_url = result[0]
+        		input_data.input,	
+        		"8-Step",	
+        		api_name="/generate_image"
+            )
+            image_url = result
             print(image_url)
             with open(image_url, 'rb') as img_file:
                 img_binary = img_file.read()