hasanmustafa0503 commited on
Commit
35d4f87
·
verified ·
1 Parent(s): d9f8da2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -1,10 +1,6 @@
1
- import gradio as gr
2
  from transformers import pipeline
3
 
4
- classifier = pipeline("text-classification", model="hasanmustafa0503/SentimentModel")
5
 
6
- def predict(text):
7
- return classifier(text)
8
-
9
- iface = gr.Interface(fn=predict, inputs="text", outputs="json")
10
- iface.launch()
 
 
1
  from transformers import pipeline
2
 
3
+ sentiment_pipeline = pipeline("text-classification", model="hasanmustafa0503/SentimentModel")
4
 
5
+ result = sentiment_pipeline("I love this!")
6
+ print(result)