“Update”

src/img_gen.py

@@ -49,11 +49,11 @@ def generate_image(prompt_alias, team, model_alias, custom_prompt, height=360, w
     # Initialize the InferenceClient
     try:
         client = InferenceClient(model_name, token=api_token)
-    #except Exception as e:
-    #    return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
+    except Exception as e:
+        return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
 
-    # Generate the image
-    #try:
+     #Generate the image
+    try:
         image = client.text_to_image(
             prompt,
             guidance_scale=guidance_scale,