wissemkarous commited on
Commit
81f670d
·
verified ·
1 Parent(s): bca7cb7
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,9 +1,7 @@
1
  # Use a pipeline as a high-level helper
2
  from transformers import pipeline, Conversation
3
  import gradio as gr
4
- pipe = pipeline("text-classification", model="UholoDala/sentence_sentiments_analysis_distilbert")
5
- # we are going to use gradio here
6
-
7
  def my_function(text):
8
  out = pipe(text)
9
  out = out[0]["label"] # cleaning the output
 
1
  # Use a pipeline as a high-level helper
2
  from transformers import pipeline, Conversation
3
  import gradio as gr
4
+ pipe = pipeline("text-classification", model="Junr-syl/sentiments_analysis_DISTILBERT")
 
 
5
  def my_function(text):
6
  out = pipe(text)
7
  out = out[0]["label"] # cleaning the output