Update app.py

app.py

@@ -19,12 +19,12 @@ pipeline = AutoPipelineForText2Image.from_pretrained(
 def read_root():
     return {"Hello": "World!"}
 
-@app.get('/image')
-def generate_image(query:str):
-    prompt = query
+@app.get('/image/{prompt}')
+def generate_image(prompt):
     print(prompt)
   
     image = pipeline(prompt=prompt, num_inference_steps=10).images[0]
+    print("Saving image")
     image.save("image_generated.png")