kwanpon commited on
Commit
afa30ff
·
verified ·
1 Parent(s): 3baf600

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -44,6 +44,11 @@ trainer = Trainer(
44
  )
45
  trainer.train()
46
 
 
 
 
 
 
47
  # inference function for gradio
48
  def classify(text):
49
  inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
 
44
  )
45
  trainer.train()
46
 
47
+ from huggingface_hub import HfApi
48
+
49
+ model.push_to_hub("kwanpon/mdeberta-classify-thai")
50
+ tokenizer.push_to_hub("kwanpon/mdeberta-classify-thai")
51
+
52
  # inference function for gradio
53
  def classify(text):
54
  inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)