Hugging Face's logo

Spaces:
TIGER-Lab
/
GenAI-Arena
Running on Zero

App Files Community
18
tianleliphoebe commited on
Commit
07e4294
·
1 Parent(s): e049190

fix

Browse files
Files changed (1) hide show
  1. model/models/fal_api_models.py +1 -1
model/models/fal_api_models.py CHANGED
@@ -44,7 +44,7 @@ class FalModel():
44
  result = handler.get()
45
  print(result)
46
  result_url = result['images'][0]['url']
47
- if self.model_name in ["SDXLTurbo", "LCM"]:
48
  result_url = io.BytesIO(decode_data_url(result_url))
49
  result = Image.open(result_url)
50
  else:
 
44
  result = handler.get()
45
  print(result)
46
  result_url = result['images'][0]['url']
47
+ if self.model_name in ["SDXLTurbo", "LCM(v1.5/XL)"]:
48
  result_url = io.BytesIO(decode_data_url(result_url))
49
  result = Image.open(result_url)
50
  else: