Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,8 @@ try:
|
|
| 35 |
print(f"✅ Trained model '{MODEL_ID}' loaded successfully.")
|
| 36 |
except Exception:
|
| 37 |
print(f"❌ Could not load trained model '{MODEL_ID}'. Loading base model.")
|
| 38 |
-
pipe = DiffusionPipeline.from_pretrained
|
|
|
|
| 39 |
|
| 40 |
pipe = pipe.to(DEVICE)
|
| 41 |
|
|
|
|
| 35 |
print(f"✅ Trained model '{MODEL_ID}' loaded successfully.")
|
| 36 |
except Exception:
|
| 37 |
print(f"❌ Could not load trained model '{MODEL_ID}'. Loading base model.")
|
| 38 |
+
pipe = DiffusionPipeline.from_pretrained
|
| 39 |
+
(BASE_MODEL_ID, torch_dtype=DTYPE, safety_checker=None)
|
| 40 |
|
| 41 |
pipe = pipe.to(DEVICE)
|
| 42 |
|