Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 
 # Define the function to query Hugging Face API for image generation
 def generate_image(prompt):
-    API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
+    API_URL = "https://api-inference.huggingface.co/models/KingNish/flux-me"
     API_TOKEN = os.getenv("HF_READ_TOKEN")  # Ensure the token is set in your environment
     headers = {"Authorization": f"Bearer {API_TOKEN}"}