Update app.py

app.py

@@ -45,24 +45,10 @@ 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']
-        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
+        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']
         
     if task == "OpenDalle v1.1":
         opendalle = str(os.getenv("opendalle"))