Hugging Face's logo

Spaces:
akhaliq
/
CLIP_prefix_captioning
Runtime error

App Files Community
1
Ahsen Khaliq commited on
Commit
f9bf321
•
1 Parent(s): d57300d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ TA = Union[T, ARRAY]
31
  D = torch.device
32
  CPU = torch.device('cpu')
33
 
34
- model_path = 'model_wieghts.pt'
35
 
36
  def get_device(device_id: int) -> D:
37
  if not torch.cuda.is_available():
 
31
  D = torch.device
32
  CPU = torch.device('cpu')
33
 
34
+ model_path = 'conceptual_weights.pt'
35
 
36
  def get_device(device_id: int) -> D:
37
  if not torch.cuda.is_available():