Spaces:
Running
on
Zero
Running
on
Zero
Avijit Ghosh
commited on
Commit
·
799c666
1
Parent(s):
346cb40
fix repeat loads
Browse files
app.py
CHANGED
|
@@ -65,8 +65,6 @@ pipeline_text2image = load_model(default_model)
|
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
def getimgen(prompt, model_name):
|
| 68 |
-
global pipeline_text2image
|
| 69 |
-
pipeline_text2image = load_model(model_name)
|
| 70 |
if model_name == "stabilityai/sdxl-turbo":
|
| 71 |
return pipeline_text2image(prompt=prompt, guidance_scale=0.0, num_inference_steps=2).images[0]
|
| 72 |
elif model_name == "runwayml/stable-diffusion-v1-5":
|
|
|
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
def getimgen(prompt, model_name):
|
|
|
|
|
|
|
| 68 |
if model_name == "stabilityai/sdxl-turbo":
|
| 69 |
return pipeline_text2image(prompt=prompt, guidance_scale=0.0, num_inference_steps=2).images[0]
|
| 70 |
elif model_name == "runwayml/stable-diffusion-v1-5":
|