Update main.py

main.py

@@ -470,10 +470,12 @@ def generate_image(request: Request, input_data: InputImage):
             		"8-Step",	
             		api_name="/generate_image"
                 )
+                image_url = result
             else: 
-                client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/9kuov/")
+                #client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/9kuov/")
+                client = Client("https://choimirai-playground-v2-5.hf.space/--replicas/bgsav/")
                 result = client.predict(
-                    input_data.input,	# str  in 'Prompt' Textbox component
+                    input_data.input,	# str  in 'Prompt' Textbox component                    
                     input_data.negativePrompt,	# str  in 'Negative prompt' Textbox component
                     True,	# bool  in 'Use negative prompt' Checkbox component
                     0,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
@@ -483,7 +485,7 @@ def generate_image(request: Request, input_data: InputImage):
                     True,	# bool  in 'Randomize seed' Checkbox component
                     api_name="/run"
                 )
-            image_url = result
+                image_url = result.image
             print(image_url)
             with open(image_url, 'rb') as img_file:
                 img_binary = img_file.read()