Spaces:
Sleeping
Sleeping
Commit ·
2853ffc
1
Parent(s): 84b5521
debug
Browse files
app.py
CHANGED
|
@@ -147,6 +147,7 @@ class Handler:
|
|
| 147 |
self.revq = ReVQ.from_pretrained("AndyRaoTHU/revq-512T")
|
| 148 |
self.revq.to(self.device)
|
| 149 |
self.revq.eval()
|
|
|
|
| 150 |
|
| 151 |
def tensor_to_image(self, tensor):
|
| 152 |
img = tensor.squeeze(0).cpu().permute(1, 2, 0).numpy()
|
|
|
|
| 147 |
self.revq = ReVQ.from_pretrained("AndyRaoTHU/revq-512T")
|
| 148 |
self.revq.to(self.device)
|
| 149 |
self.revq.eval()
|
| 150 |
+
print("Models loaded successfully!")
|
| 151 |
|
| 152 |
def tensor_to_image(self, tensor):
|
| 153 |
img = tensor.squeeze(0).cpu().permute(1, 2, 0).numpy()
|