SoDehghan commited on
Commit
8859c67
·
verified ·
1 Parent(s): 83179f0

Update models/hsd_ar.py

Browse files
Files changed (1) hide show
  1. models/hsd_ar.py +2 -3
models/hsd_ar.py CHANGED
@@ -5,8 +5,7 @@ import time
5
  from transformers import pipeline
6
  import os
7
 
8
- import streamlit as st
9
- from transformers import pipeline
10
 
11
  sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
12
 
@@ -21,4 +20,4 @@ if user_input:
21
  sentiment = result[0]["label"]
22
 
23
  sentiment = label_dict[sentiment]
24
- st.write(f"Sentiment: {sentiment}")
 
5
  from transformers import pipeline
6
  import os
7
 
8
+
 
9
 
10
  sentiment_pipeline = pipeline(task = "text-classification", model = "gritli/bert-sentiment-analyses-imdb")
11
 
 
20
  sentiment = result[0]["label"]
21
 
22
  sentiment = label_dict[sentiment]
23
+ st.write(f"Detection: {sentiment}")