Update generate_propmts.py

generate_propmts.py

@@ -82,24 +82,17 @@ def get_image_from_space(text, sentence_mapping, character_dict, selected_style,
         bytes: The generated image data in bytes format, or None if the request fails.
     """
     image_bytes = None  # Initialize image bytes to None for error handling
-    try:
-        with st.spinner("עוד כמה רגעים והאיור יופיע"):
-            # Define the payload with the text prompt
-            prompt,_ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
-            payload = {
-                "inputs": prompt
-            }
-            result = client.predict(
-                prompt=payload,
-                api_name="/predict"
-            )
+    # Define the payload with the text prompt
+    prompt,_ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
+    payload = {
+        "inputs": prompt
+    }
+    result = client.predict(
+        prompt=payload,
+        api_name="/predict"
+    )
 
-            # Check if the result is a base64 encoded string
-            if isinstance(result, str):
-                image_bytes = base64.b64decode(result)
-                return image_bytes
-            else:
-                st.error("אוי לא ניתן לייצר תמונה, יש לנסות שוב בהמשך")
-    except Exception as e:
-        print(f"אוי לא ניתן לייצר תמונה, יש לנסות שוב בהמשך: {e}")
-        return image_bytes
+    # Check if the result is a base64 encoded string
+    if isinstance(result, str):
+        image_bytes = base64.b64decode(result)
+    return image_bytes