Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ else:
|
|
| 53 |
dtype = torch.float32
|
| 54 |
|
| 55 |
print(f"Loading pipeline on {device} with {dtype}...")
|
| 56 |
-
|
| 57 |
model_dir,
|
| 58 |
device=device,
|
| 59 |
dtype=dtype,
|
|
|
|
| 53 |
dtype = torch.float32
|
| 54 |
|
| 55 |
print(f"Loading pipeline on {device} with {dtype}...")
|
| 56 |
+
pipe = HeartMuLaGenPipeline.from_pretrained(
|
| 57 |
model_dir,
|
| 58 |
device=device,
|
| 59 |
dtype=dtype,
|