Rename main.py to app.py
Browse files- main.py → app.py +8 -14
main.py → app.py
RENAMED
@@ -1,17 +1,8 @@
|
|
1 |
-
from
|
2 |
-
from
|
3 |
-
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
4 |
import torch
|
5 |
|
6 |
-
app =
|
7 |
-
|
8 |
-
app.add_middleware(
|
9 |
-
CORSMiddleware,
|
10 |
-
allow_origins=["*"],
|
11 |
-
allow_methods=["*"],
|
12 |
-
allow_headers=["*"],
|
13 |
-
allow_credentials=True,
|
14 |
-
)
|
15 |
|
16 |
model_id = "runwayml/stable-diffusion-v1-5"
|
17 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
@@ -22,9 +13,12 @@ def dummy(images, **kwargs):
|
|
22 |
|
23 |
pipe.safety_checker = dummy
|
24 |
|
25 |
-
@app.
|
26 |
def generate_image():
|
27 |
prompt = request.args.get('prompt')
|
28 |
image = pipe(prompt).images[0]
|
29 |
# do something with the generated image
|
30 |
-
return image
|
|
|
|
|
|
|
|
1 |
+
from flask import Flask, request
|
2 |
+
from diffusers import StableDiffusionPipeline
|
|
|
3 |
import torch
|
4 |
|
5 |
+
app = Flask(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
model_id = "runwayml/stable-diffusion-v1-5"
|
8 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
|
|
13 |
|
14 |
pipe.safety_checker = dummy
|
15 |
|
16 |
+
@app.route('/')
|
17 |
def generate_image():
|
18 |
prompt = request.args.get('prompt')
|
19 |
image = pipe(prompt).images[0]
|
20 |
# do something with the generated image
|
21 |
+
return image
|
22 |
+
|
23 |
+
if __name__ == '__main__':
|
24 |
+
app.run()
|