rviana commited on
Commit
de6e536
1 Parent(s): d5eae70

Add distilbert multilingual model.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import streamlit as st
4
  import socket
5
  import threading
6
 
7
- # Specify the model name explicitly to avoid warnings
8
- model_name = "distilbert-base-uncased"
9
 
10
  # Load the pre-trained sentiment-analysis pipeline
11
  try:
 
4
  import socket
5
  import threading
6
 
7
+ # Specify the multilingual DistilBERT model
8
+ model_name = "distilbert-base-multilingual-cased"
9
 
10
  # Load the pre-trained sentiment-analysis pipeline
11
  try: