Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import torch
-from transformers import RobertaTokenizer, RobertaForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from datasets import load_dataset, Dataset
 import pandas as pd
 
@@ -21,8 +21,8 @@ df['label'] = df['intent'].map(label2id)
 df = df[['instruction', 'label', 'intent', 'response']]
 
 # Load the tokenizer and model
-tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
-model = RobertaForSequenceClassification.from_pretrained('path_to_your_saved_model_directory')
+tokenizer = AutoTokenizer.from_pretrained("Chillyblast/Roberta_Question_Answer")
+model = AutoModelForSequenceClassification.from_pretrained("Chillyblast/Roberta_Question_Answer")
 
 # Ensure the model is in evaluation mode
 model.eval()