Update app.py

app.py

@@ -14,12 +14,12 @@ def query(payload):
     return response.content
 
 # Check for 'text' parameter in URL
-query_params = st.experimental_get_query_params()
-prompt_from_url = query_params.get('text', None)
+query_params = st.query_params
+prompt_from_url = query_params.get('text')
 
 if prompt_from_url:
     # If 'text' parameter is present, generate and display the image only
-    prompt = prompt_from_url[0]
+    prompt = prompt_from_url
     image_bytes = query({"inputs": prompt})
     image = Image.open(io.BytesIO(image_bytes))
     st.image(image, caption="Generated Image", use_container_width=True)