Kimang18 commited on
Commit
8b2ff9f
·
verified ·
1 Parent(s): b1ab0c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -13,6 +13,7 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
13
 
14
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
15
  model = TrorYongASRModel.from_pretrained(model_id)
 
16
 
17
  @torch.no_grad()
18
  def transcribe_easier(filepath):
 
13
 
14
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
15
  model = TrorYongASRModel.from_pretrained(model_id)
16
+ model.eval()
17
 
18
  @torch.no_grad()
19
  def transcribe_easier(filepath):