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]
|