Spaces:
Runtime error
Runtime error
| import json | |
| import streamlit as st | |
| from transformers import AutoTokenizer, RobertaForSequenceClassification | |
| with open("config.json") as f: | |
| cfg = json.loads(f.read()) | |
| # @st.cache(allow_output_mutation=True) | |
| # def load_model(): | |
| # tokenizer = AutoTokenizer.from_pretrained(cfg["model_name_or_path"]) | |
| # model = RobertaForSequenceClassification.from_pretrained(cfg["model_name_or_path"]) | |
| st.title("RoBERTa Marathi") | |
| # with st.spinner("Loading model..."): | |
| # generator, tokenizer = load_model() | |
| input_text = st.text_input("Text:") | |
| predict_button = st.button("Predict") | |
| if predict_button: | |
| with st.spinner("Generating prediction..."): | |
| # Get prediction here | |
| st.write("Predicted Label") | |