longvnhue1 commited on
Commit
781522e
·
1 Parent(s): e549401

Remove local model folder

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def split_by_words_and_dot(text, min_words=125, max_words=160, fallback_words=15
30
  return chunks
31
 
32
  # Load model
33
- model_path = "./model/facebook-m2m100_418M-fine_tuning"
34
  tokenizer = M2M100Tokenizer.from_pretrained(model_path)
35
  model = M2M100ForConditionalGeneration.from_pretrained(model_path)
36
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
30
  return chunks
31
 
32
  # Load model
33
+ model_path = "longvnhue1/finetune-deploy1"
34
  tokenizer = M2M100Tokenizer.from_pretrained(model_path)
35
  model = M2M100ForConditionalGeneration.from_pretrained(model_path)
36
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")