Update app.py

app.py

@@ -1,9 +1,12 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
+# Replace with your Hugging Face model repository path
+model_repo_path = 'nxmwxm/correct_answer'
+
 # Load the model and tokenizer
-model = T5ForConditionalGeneration.from_pretrained('/content/fine_tuned_model')
-tokenizer = T5Tokenizer.from_pretrained('/content/fine_tuned_model')
+model = BartForConditionalGeneration.from_pretrained(model_repo_path)
+tokenizer = BartTokenizer.from_pretrained(model_repo_path)
 
 def generate_answer(question, distractors):
     input_text = f"Question: {question} Distractors: {' '.join(distractors)}"