Update main.py
Browse files
main.py
CHANGED
@@ -25,6 +25,10 @@ def dummy(images, **kwargs):
|
|
25 |
pipe.safety_checker = dummy
|
26 |
|
27 |
@app.route('/')
|
|
|
|
|
|
|
|
|
28 |
def generate_image():
|
29 |
prompt = request.args.get('prompt')
|
30 |
image = pipe(prompt).images[0]
|
|
|
25 |
pipe.safety_checker = dummy
|
26 |
|
27 |
@app.route('/')
|
28 |
+
def hello():
|
29 |
+
return "Hello Art"
|
30 |
+
|
31 |
+
@app.route('/gen')
|
32 |
def generate_image():
|
33 |
prompt = request.args.get('prompt')
|
34 |
image = pipe(prompt).images[0]
|