Update app.py

app.py

@@ -48,7 +48,7 @@ def load_generated_images():
     
     # Load all images from the directory
     image_files = [os.path.join(SAVE_DIR, f) for f in os.listdir(SAVE_DIR) 
-                   if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
+                  if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
     # Sort by creation time (newest first)
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
@@ -58,7 +58,7 @@ def load_predefined_images():
     return []
 
 @spaces.GPU(duration=120)
-def predict(
+def inference(
     prompt: str,
     seed: int,
     randomize_seed: bool,
@@ -109,11 +109,10 @@ footer {
 }
 """
 
-# Create Gradio Blocks with API documentation disabled (show_api=False)
-with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=False) as demo:
+with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=False) as demo:
     gr.HTML('<div class="title"> FLUX Ghibli LoRA</div>')
     gr.HTML('<div class="title">😄Image to Video Explore: <a href="https://huggingface.co/spaces/ginigen/theater" target="_blank">https://huggingface.co/spaces/ginigen/theater</a></div>')
-    
+
     with gr.Tabs() as tabs:
         with gr.Tab("Generation"):
             with gr.Column(elem_id="col-container"):
@@ -126,9 +125,9 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
                         container=False,
                     )
                     run_button = gr.Button("Run", scale=0)
-                
+
                 result = gr.Image(label="Result", show_label=False)
-                
+
                 with gr.Accordion("Advanced Settings", open=False):
                     seed = gr.Slider(
                         label="Seed",
@@ -138,7 +137,7 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
                         value=42,
                     )
                     randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-                    
+
                     with gr.Row():
                         width = gr.Slider(
                             label="Width",
@@ -154,7 +153,7 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
                             step=32,
                             value=768,
                         )
-                    
+
                     with gr.Row():
                         guidance_scale = gr.Slider(
                             label="Guidance scale",
@@ -177,13 +176,13 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
                             step=0.1,
                             value=1.0,
                         )
-                
+
                 gr.Examples(
                     examples=examples,
                     inputs=[prompt],
                     outputs=[result, seed],
                 )
-        
+
         with gr.Tab("Gallery"):
             gallery_header = gr.Markdown("### Generated Images Gallery")
             generated_gallery = gr.Gallery(
@@ -195,20 +194,21 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
                 height="auto"
             )
             refresh_btn = gr.Button("🔄 Refresh Gallery")
-    
-    # Event handler to refresh gallery
+
+
+    # Event handlers
     def refresh_gallery():
         return load_generated_images()
-    
+
     refresh_btn.click(
         fn=refresh_gallery,
         inputs=None,
         outputs=generated_gallery,
     )
-    
+
     gr.on(
         triggers=[run_button.click, prompt.submit],
-        fn=predict,
+        fn=inference,
         inputs=[
             prompt,
             seed,
@@ -223,4 +223,4 @@ with gr.Blocks(show_api=False, theme="Yntec/HaleyCH_Theme_Orange", css=css, anal
     )
 
 demo.queue()
-demo.launch()  # Removed share=True for Spaces
+demo.launch()