Upload folder using huggingface_hub

app.py

@@ -268,7 +268,7 @@ def create_error_image(backend, error_message):
         img_bytes = buffer.getvalue()
 
         # Convert to base64 and return as data URI
-        return f"data:image/png;base64,{base64.b64encode(img_bytes).decode('utf-8')}"
+        return f"data:image/jpeg;base64,{base64.b64encode(img_bytes).decode('utf-8')}"
     except Exception as e:
         print(f"Failed to create error image: {e}")
         # Return a simple error message as fallback
@@ -333,7 +333,7 @@ async def poll_once(manager, backend, request_id):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     session_id = gr.State(None)  # Add this to store session ID
 
-    gr.Markdown("# 🌊 WaveSpeed AI Image Generator")
+    gr.Markdown("# 🌊 HiDream Arena powered by WaveSpeed AI Image Generator")
 
     # Add the introduction with link to WaveSpeedAI
     gr.Markdown(

requirements.txt

@@ -2,3 +2,4 @@ gradio
 aiohttp
 plotly
 python-dotenv
+pydantic==2.8.2