| from datasets import load_dataset | |
| from setfit import SetFitModel, Trainer, TrainingArguments, sample_dataset | |
| import streamlit as st | |
| fine_tuned_model = SetFitModel.from_pretrained("./my-setfit-finetuned-model") | |
| st.title("This is a sentiment analysis of a sentence") | |
| title = st.text_input("Sentiment of sentence", "I hate this") | |
| new_preds = fine_tuned_model.predict(title) | |
| st.write("The current sentiment of this sentence is", new_preds) | |