cdactvm commited on
Commit
153933b
·
verified ·
1 Parent(s): 82814b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ original_model=Wav2Vec2BertForCTC.from_pretrained(model_name)
14
  # Explicitly allow Wav2Vec2BertForCTC during unpickling3+
15
  torch.serialization.add_safe_globals([Wav2Vec2BertForCTC])
16
  # Load the full quantized model
17
- quantized_model = torch.load("model_name", weights_only=False)
18
  quantized_model.eval()
19
 
20
  #####################################################
 
14
  # Explicitly allow Wav2Vec2BertForCTC during unpickling3+
15
  torch.serialization.add_safe_globals([Wav2Vec2BertForCTC])
16
  # Load the full quantized model
17
+ quantized_model = torch.load("cdactvm/w2v-bert-punjabi/wav2vec2_bert_qint8.pth", weights_only=False)
18
  quantized_model.eval()
19
 
20
  #####################################################