Update main.py

main.py

@@ -24,14 +24,13 @@ def dummy(images, **kwargs):
 
 pipe.safety_checker = dummy
 
-@app.route('/')
+@app.get("/")
 def hello():
     return "Hello, I'm Linlada"
 
 
-@app.route('/gen')
-def generate_image():
-    prompt = request.args.get('prompt')
+@app.get("/gen/{prompt}")
+def generate_image(prompt: str):
     image = pipe(prompt).images[0]
     # Save the image
     image.save('static/image.png')