AndyRaoTHU commited on
Commit
2853ffc
·
1 Parent(s): 84b5521
Files changed (1) hide show
  1. app.py +1 -0
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()