CR7CAD commited on
Commit
8021e09
·
verified ·
1 Parent(s): 19a0df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -66,16 +66,16 @@ def load_models():
66
  # Use pipeline if available
67
  models['evaluator'] = pipeline(
68
  "sentiment-analysis",
69
- model="cardiffnlp/twitter-roberta-base-sentiment-latest"
70
  )
71
  else:
72
  # Fall back to basic model loading using the updated evaluation model
73
  try:
74
  models['evaluator_model'] = AutoModelForSequenceClassification.from_pretrained(
75
- "cardiffnlp/twitter-roberta-base-sentiment-latest"
76
  )
77
  models['evaluator_tokenizer'] = AutoTokenizer.from_pretrained(
78
- "cardiffnlp/twitter-roberta-base-sentiment-latest"
79
  )
80
  except Exception as e:
81
  st.error(f"Error loading sentiment model: {e}")
 
66
  # Use pipeline if available
67
  models['evaluator'] = pipeline(
68
  "sentiment-analysis",
69
+ model="CR7CAD/RobertaFinetuned"
70
  )
71
  else:
72
  # Fall back to basic model loading using the updated evaluation model
73
  try:
74
  models['evaluator_model'] = AutoModelForSequenceClassification.from_pretrained(
75
+ "CR7CAD/RobertaFinetuned"
76
  )
77
  models['evaluator_tokenizer'] = AutoTokenizer.from_pretrained(
78
+ "CR7CAD/RobertaFinetuned"
79
  )
80
  except Exception as e:
81
  st.error(f"Error loading sentiment model: {e}")