Changed from Named Entity Recognition (NER) to text-to-image

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 hf_api_key = os.environ['HF_API_KEY']
 
 # Text-to-image endpoint
-get_completion = pipeline("ner", model="sd-legacy/stable-diffusion-v1-5")
+get_completion = pipeline("text-to-image", model="sd-legacy/stable-diffusion-v1-5")
 
 
 # A helper function to convert the PIL image to base64,