Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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']
|