Spaces:

Aananda-giri
/

gpt2-nepali

Running

Aananda-Giri commited on 29 days ago

Commit

4e082c0

1 Parent(s): f51e14f

remove value

Files changed (1) hide show

app.py CHANGED Viewed

@@ -220,7 +220,7 @@ def load_model_n_tokenizer():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device)
-    tokenizer = PreTrainedTokenizerFast.from_pretrained("Aananda-giri/NepaliBPE")
     return model, tokenizer
 def generate(

     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device)
+    tokenizer = PreTrainedTokenizerFast.from_pretrained("Aananda-giri/GPT2-Nepali")
     return model, tokenizer
 def generate(