aminskjen commited on
Commit
2722687
·
verified ·
1 Parent(s): a5ce6ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -192,12 +192,18 @@ def generate_image_hf(scene_idea, story):
192
  json={"inputs": prompt},
193
  timeout=15
194
  )
195
- logging.info(f"model: {model}, status:{res.status_code},response: {res.text[:300]}")
196
  if res.status_code == 200:
197
  import base64
198
  return f"data:image/png;base64,{base64.b64encode(res.content).decode()}"
 
 
 
 
199
  except Exception as e:
 
200
  continue
 
201
  return generate_svg_placeholder(scene_idea, story)
202
 
203
  def generate_svg_placeholder(scene_idea, story):
 
192
  json={"inputs": prompt},
193
  timeout=15
194
  )
195
+ logging.info(f"Status: {res.status_code}")
196
  if res.status_code == 200:
197
  import base64
198
  return f"data:image/png;base64,{base64.b64encode(res.content).decode()}"
199
+
200
+ else:
201
+ logging.warning(f"Model {model} failed: {res.status_code} - {res.text[:200]}")
202
+
203
  except Exception as e:
204
+ logging.error(f"Error from model {model}: {e}")
205
  continue
206
+
207
  return generate_svg_placeholder(scene_idea, story)
208
 
209
  def generate_svg_placeholder(scene_idea, story):