rviana commited on
Commit
924dfb2
·
1 Parent(s): 3722485

Add Model.

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,16 +1,13 @@
1
  import gradio as gr
 
2
  import streamlit as st
3
  import socket
4
 
5
- try:
6
- from transformers import pipeline
7
- except ImportError as e:
8
- st.error(f"ImportError: {e}")
9
- st.stop()
10
 
11
- # Load the pre-trained sentiment-analysis pipeline
12
  try:
13
- classifier = pipeline('sentiment-analysis')
14
  except Exception as e:
15
  st.error(f"Error loading pipeline: {e}")
16
  st.stop()
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
  import streamlit as st
4
  import socket
5
 
6
+ # Specify the model name explicitly to avoid warnings
7
+ model_name = "distilbert-base-uncased-finetuned-sst-2-english"
 
 
 
8
 
 
9
  try:
10
+ classifier = pipeline('sentiment-analysis', model=model_name)
11
  except Exception as e:
12
  st.error(f"Error loading pipeline: {e}")
13
  st.stop()