Diary14 commited on
Commit
4717ec6
·
verified ·
1 Parent(s): a069696

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -3,15 +3,15 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
4
  import pandas as pd
5
 
6
- st.set_page_config(page_title="Analyse de Sentiment en Lot", page_icon="📊", layout="wide")
7
 
8
- st.title("📊 Analyse de Sentiment Multiline (Batch Processing)")
9
  st.write("Collez un ou plusieurs avis (un par ligne) pour les analyser simultanément avec RoBERTa.")
10
 
11
  # Chargement du modèle mis en cache
12
  @st.cache_resource
13
  def load_model():
14
- model_id = "TON_PSEUDO_HF/TON_MODELE" # ⚠️ À remplacer
15
  tokenizer = AutoTokenizer.from_pretrained(model_id)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_id)
17
  return tokenizer, model
 
3
  import torch
4
  import pandas as pd
5
 
6
+ st.set_page_config(page_title="Analyse de Sentiment", layout="wide")
7
 
8
+ st.title(" Analyse de Sentiment")
9
  st.write("Collez un ou plusieurs avis (un par ligne) pour les analyser simultanément avec RoBERTa.")
10
 
11
  # Chargement du modèle mis en cache
12
  @st.cache_resource
13
  def load_model():
14
+ model_id = "Diary14/roberta-sentiment-lora" # ⚠️ À remplacer
15
  tokenizer = AutoTokenizer.from_pretrained(model_id)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_id)
17
  return tokenizer, model