Update main.py

main.py

@@ -28,7 +28,7 @@ pipe.safety_checker = dummy
 
 @app.get("/")
 def hello():
-    return "Hello, I'm Linlada"
+    return "Hello, I'm Artist"
 
 
 @app.get("/gen/{prompt}")
@@ -37,10 +37,7 @@ def generate_image(prompt: str):
                   guidance_scale=8.5  # how strict to follow the prompt
                   ).images[0]
     # Save the image
-    image.save('static/image.png')
+    image.save('image.png')
 
-    # do something with the generated image
-    image_data = image.tobytes().hex()
-    image_url = url_for('static', filename='image.png')
 
-    return {'image_data': image_data, 'image_url': image_url}
+    return "Gen done"