Update app.py

app.py

@@ -45,11 +45,25 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     url_sd6 = os.getenv("url_sd6")
     hf_token = os.getenv("hf_token")
     if task == "Playground v2":
-        playground = str(os.getenv("playground"))
-        client = Client(playground, hf_token=hf_token)
-        result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
-        return result[0][0]['image']
-
+        #playground = str(os.getenv("playground"))
+        #client = Client(playground, hf_token=hf_token)
+        #result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
+        #return result[0][0]['image']
+        client = Client("multimodalart/stable-cascade")
+        result = client.predict(
+            "cat",	# str  in 'Prompt' Textbox component
+            "",	# str  in 'Negative prompt' Textbox component
+            0,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
+            1024,	# float (numeric value between 1024 and 1536) in 'Width' Slider component
+            1024,	# float (numeric value between 1024 and 1536) in 'Height' Slider component
+            20,	# float (numeric value between 10 and 30) in 'Prior Inference Steps' Slider component
+            4,	# float (numeric value between 0 and 20) in 'Prior Guidance Scale' Slider component
+            10,	# float (numeric value between 4 and 12) in 'Decoder Inference Steps' Slider component
+            0,	# float (numeric value between 0 and 0) in 'Decoder Guidance Scale' Slider component
+            1,	# float (numeric value between 1 and 2) in 'Number of Images' Slider component
+		api_name="/run")
+        return result
+        
     if task == "OpenDalle v1.1":
         opendalle = str(os.getenv("opendalle"))
         client = Client(opendalle, hf_token=hf_token)