Update app.py

app.py

@@ -6,7 +6,6 @@ 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 = """
@@ -86,10 +85,8 @@ async def api_generate(request: Request):
         "image_format": "jpeg"
     }
 
-# Create Gradio app
-with gr.Blocks(css=css_code, title="Image Generation App") as demo: 
-    gr.Markdown("# Image Generation Tool")
-    
+# Gradio Interface
+def make_me():
     with gr.Row():
         # Left Column (50% width)
         with gr.Column(scale=1, min_width=400):
@@ -135,6 +132,7 @@ with gr.Blocks(css=css_code, title="Image Generation App") as demo:
                 container=False
             )
             
+    # Functionality remains the same
     def generate_wrapper(model_str, prompt):
         # Translate prompt to English
         translated_prompt = translator.translate(prompt)
@@ -180,11 +178,15 @@ with gr.Blocks(css=css_code, title="Image Generation App") as demo:
         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__":