tomyswall commited on
Commit
3d3b40e
1 Parent(s): 12bb76c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
3
 
4
- model = AutoModelForSequenceClassification.from_pretrained("savasy/bert-base-turkish-sentiment-cased")
5
- tokenizer = AutoTokenizer.from_pretrained("savasy/bert-base-turkish-sentiment-cased")
6
 
7
  def greet(text):
8
  pipe = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)
 
1
  import gradio as gr
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
3
 
4
+ model = AutoModelForSequenceClassification.from_pretrained("MrHup/coloring-book")
5
+ tokenizer = AutoTokenizer.from_pretrained("MrHup/coloring-book")
6
 
7
  def greet(text):
8
  pipe = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)