Upload folder using huggingface_hub

app.py

@@ -17,8 +17,7 @@ def generate_image(prompt, progress=gr.Progress()):
         progress(1.0, desc="Done!")
         return image
     except Exception as e:
-        print(f"Error: {e}")
-        return None
+        raise gr.Error(f"FLUX Inference Failed: {str(e)}")
 
 # A sleek Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -36,7 +35,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     ["A futuristic city built into a giant tree, cinematic lighting, conceptual art"], 
                     ["An astronaut riding a glowing translucent horse on the surface of Mars, 8k resolution"]
                 ],
-                inputs=prompt
+                inputs=[prompt]
             )
         with gr.Column(scale=1):
             output = gr.Image(label="Generated Artwork", type="pil")