dhmeltzer commited on
Commit
7e36539
·
1 Parent(s): cffb37c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -10,13 +10,11 @@ def main():
10
  Simply input a review into the text below and the application will give two predictions for what the \
11
  rating is on a scale of 0-5. The models will also produce the score they assigned their prediction. The score is\
12
  between 0 and 1 and quantifies the confidence the model has in its prediction.\
13
- \n\n \
14
- Specifically, we consider two pre-trained models, [BERT-tiny](https://huggingface.co/dhmeltzer/bert-tiny-goodreads-wandb) and [DistilBERT](https://huggingface.co/dhmeltzer/distilbert-goodreads-wandb)\
15
  which have been fine-tuned on a dataset of Goodreads book \
16
  reviews, see [here](https://www.kaggle.com/competitions/goodreads-books-reviews-290312/data) for the original dataset. \
17
  These models are deployed on AWS and are accessed using a REST API. To deploy the models we used a combination of AWS Sagemaker, Lambda, and API Gateway.\
18
- \n\n \
19
- To read more about this project and specifically how we cleaned the data and trained the models, see the following GitHub (repository)[https://github.com/david-meltzer/Goodreads-Sentiment-Analysis].")
20
 
21
 
22
  AWS_key = st.secrets['AWS-key']
 
10
  Simply input a review into the text below and the application will give two predictions for what the \
11
  rating is on a scale of 0-5. The models will also produce the score they assigned their prediction. The score is\
12
  between 0 and 1 and quantifies the confidence the model has in its prediction.\
13
+ \n\n Specifically, we consider two pre-trained models, [BERT-tiny](https://huggingface.co/dhmeltzer/bert-tiny-goodreads-wandb) and [DistilBERT](https://huggingface.co/dhmeltzer/distilbert-goodreads-wandb)\
 
14
  which have been fine-tuned on a dataset of Goodreads book \
15
  reviews, see [here](https://www.kaggle.com/competitions/goodreads-books-reviews-290312/data) for the original dataset. \
16
  These models are deployed on AWS and are accessed using a REST API. To deploy the models we used a combination of AWS Sagemaker, Lambda, and API Gateway.\
17
+ \n\n To read more about this project and specifically how we cleaned the data and trained the models, see the following GitHub (repository)[https://github.com/david-meltzer/Goodreads-Sentiment-Analysis].")
 
18
 
19
 
20
  AWS_key = st.secrets['AWS-key']