Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -16,9 +16,11 @@ def reload_model(new_model_id):
|
|
16 |
global pipe, model_id, prev_height, prev_width
|
17 |
model_id = new_model_id
|
18 |
try:
|
19 |
-
|
|
|
20 |
pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
|
21 |
pipe.compile()
|
|
|
22 |
return f"Model successfully loaded: {model_id}"
|
23 |
except Exception as e:
|
24 |
return f"Failed to load model: {str(e)}"
|
|
|
16 |
global pipe, model_id, prev_height, prev_width
|
17 |
model_id = new_model_id
|
18 |
try:
|
19 |
+
print(f"{model_id}...")
|
20 |
+
pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
|
21 |
pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
|
22 |
pipe.compile()
|
23 |
+
print(f"{model_id}!!!")
|
24 |
return f"Model successfully loaded: {model_id}"
|
25 |
except Exception as e:
|
26 |
return f"Failed to load model: {str(e)}"
|