Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Replace with your Hugging Face model repository path
-model_repo_path = 'Muh113/bart-large'
+model_repo_path = 'facebook/bart-large'
 
 # Check for GPU availability and set device accordingly
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -27,6 +27,7 @@ if st.button("Get Answer"):
                 inputs = tokenizer(question_input, return_tensors="pt", truncation=True, max_length=116).to(device)
                 # Generate the answer
                 outputs = model.generate(inputs['input_ids'], max_length=150, num_beams=4, early_stopping=True)
+                # Decode the generated answer
                 answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
                 st.subheader("Answer")
                 st.write(answer)