Update app.py

app.py

@@ -39,10 +39,11 @@ class ImageStoryteller:
                 llm_model_id,
                 torch_dtype=torch.float16,
                 device_map="auto",
+                trust_remote_code=True)
                 # trust_remote_code=True if "phi" in llm_model_id else False
                 # To this (for Qwen and other models):
-                trust_remote_code=True if any(keyword in llm_model_id.lower() for keyword in ["phi", "qwen", "yi", "deepseek"]) else False
-            )
+                # trust_remote_code=True if any(keyword in llm_model_id.lower() for keyword in ["phi", "qwen", "yi", "deepseek"]) else False
+
             print(f"LLM model {llm_model_id} loaded successfully!")
         except Exception as e:
             print(f"LLM loading failed: {e}")