Pendrokar commited on
Commit
89bf2ed
·
verified ·
1 Parent(s): 288d6f0

cloned model#2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import torch
3
  from transformers import pipeline, AutoTokenizer, DistilBertForSequenceClassification
4
 
5
- modelName = "colinryan/hf-deepmoji"
6
 
7
  #distil_tokenizer = AutoTokenizer.from_pretrained(modelName)
8
  #distil_tokenizer.save_pretrained("./model/")
 
2
  import torch
3
  from transformers import pipeline, AutoTokenizer, DistilBertForSequenceClassification
4
 
5
+ modelName = "Pendrokar/TorchMoji"
6
 
7
  #distil_tokenizer = AutoTokenizer.from_pretrained(modelName)
8
  #distil_tokenizer.save_pretrained("./model/")