mrgmd01 commited on
Commit
03b573d
·
verified ·
1 Parent(s): 21a70c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,17 +7,17 @@ import re
7
  # Load models
8
  english_model = pipeline(
9
  "sentiment-analysis",
10
- model="cardiffnlp/twitter-roberta-base-sentiment-latest"
11
  )
12
 
13
  urdu_model = pipeline(
14
  "sentiment-analysis",
15
- model="mrgmd01/sentiment_model_FineTune_cardiffnlp"
16
  )
17
 
18
  roman_urdu_model = pipeline(
19
  "sentiment-analysis",
20
- model="mrgmd01/sentiment_model_FineTune_cardiffnlp" # replace with roman urdu model if available
21
  )
22
 
23
  # File to store only sentences
 
7
  # Load models
8
  english_model = pipeline(
9
  "sentiment-analysis",
10
+ model="siebert/sentiment-roberta-large-english"
11
  )
12
 
13
  urdu_model = pipeline(
14
  "sentiment-analysis",
15
+ model="mrgmd01/SA_Model_bert-base-multilingual-uncased"
16
  )
17
 
18
  roman_urdu_model = pipeline(
19
  "sentiment-analysis",
20
+ model="mrgmd01/SA_Model_bert-base-multilingual-uncased" # replace with roman urdu model if available
21
  )
22
 
23
  # File to store only sentences