Hugging Face's logo

Spaces:
DuongTrongChi
/
Toxic_Comment_Classification
Sleeping

App Files Community
DuongTrongChi commited on
Commit
e3c1518
·
verified ·
1 Parent(s): 04a19cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ class ToxicCommentClassification:
29
 
30
 
31
  def main():
32
- model = ToxicCommentClassification("DuongTrongChi/facebook-commet-classification-base")
33
  iface = gr.Interface(
34
  fn=model.predict,
35
  inputs=gr.Textbox(
 
29
 
30
 
31
  def main():
32
+ model = ToxicCommentClassification("DuongTrongChi/facebook-commet-classification-small")
33
  iface = gr.Interface(
34
  fn=model.predict,
35
  inputs=gr.Textbox(