Spaces:

Preetham04
/

clothing_bert

Sleeping

Preetham04 commited on Jul 14

Commit

02f956c

•

1 Parent(s): b7bc321

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-pipeline = pipeline(task="text-classification", model="Preetham04/sentiment-analysis")
 def predict(input_text):
     predictions = pipeline(input_text)

 import gradio as gr
 from transformers import pipeline
+pipeline = pipeline(task="text-classification", model="Preetham04/sentiment-analysis_tf")
 def predict(input_text):
     predictions = pipeline(input_text)