Update app.py

app.py

@@ -6,6 +6,7 @@ from all_models import models
 from io import BytesIO
 from PIL import Image
 from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
 from deep_translator import GoogleTranslator
 
 css_code = """
@@ -85,8 +86,10 @@ async def api_generate(request: Request):
         "image_format": "jpeg"
     }
 
-# Gradio Interface
-def make_me():
+# Create Gradio app
+with gr.Blocks(css=css_code, title="Image Generation App") as demo: 
+    gr.Markdown("# Image Generation Tool")
+    
     with gr.Row():
         # Left Column (50% width)
         with gr.Column(scale=1, min_width=400):
@@ -132,7 +135,6 @@ def make_me():
                 container=False
             )
             
-    # Functionality remains the same
     def generate_wrapper(model_str, prompt):
         # Translate prompt to English
         translated_prompt = translator.translate(prompt)
@@ -178,15 +180,11 @@ def make_me():
         cancels=[gen_event]
     )
 
-# Create Gradio app
-with gr.Blocks(css=css_code, title="Image Generation App") as demo: 
-    gr.Markdown("# Image Generation Tool")
-    make_me()
-
 # Enable queue before mounting
 demo.queue()
 
 # Mount Gradio app to FastAPI
+app.mount("/static", StaticFiles(directory="static"), name="static")
 app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":