Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import spaces
-import random
+from huggingface_hub import HfApi
 
 @spaces.GPU(duration=60)
 def generate_image(prompt, negative_prompt="", seed=-1, width=512, height=512, guidance_scale=7.5, num_inference_steps=50):
@@ -33,7 +33,7 @@ with gr.Blocks() as demo:
     
     # Set up the function call using the loaded Flux model interface
     generate_button.click(
-        flux_demo,
+        flux_demo.predict,
         inputs=[prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps],
         outputs=[image_output, seed_output]
     )