Update app.py

app.py

@@ -183,12 +183,14 @@ def generate_image_hf(scene_idea, story):
         "stabilityai/stable-diffusion-2-1",
         "CompVis/stable-diffusion-v1-4"
     ]:
-        try:
-            res = requests.post(
+        for model in model:
+            try:
+                logging.info(f"trying model: {model}")
+                res = requests.post(
                 f"https://api-inference.huggingface.co/models/{model}",
                 headers={"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"},
                 json={"inputs": prompt},
-                timeout=60
+                timeout=15
             )
             logging.info(f"model: {model}, status:{res.status_code},response: {res.text[:300]}")
             if res.status_code == 200: