Spaces:

TIGER-Lab
/

GenAI-Arena

Running on Zero

tianleliphoebe commited on May 2

Commit

07e4294

•

1 Parent(s): e049190

fix

Files changed (1) hide show

model/models/fal_api_models.py CHANGED Viewed

@@ -44,7 +44,7 @@ class FalModel():
             result = handler.get()
             print(result)
             result_url = result['images'][0]['url']
-            if self.model_name in ["SDXLTurbo", "LCM"]:
                 result_url = io.BytesIO(decode_data_url(result_url))
                 result = Image.open(result_url)
             else:

             result = handler.get()
             print(result)
             result_url = result['images'][0]['url']
+            if self.model_name in ["SDXLTurbo", "LCM(v1.5/XL)"]:
                 result_url = io.BytesIO(decode_data_url(result_url))
                 result = Image.open(result_url)
             else: