Hugging Face's logo

Spaces:
GSridhar1982
/
AIML_QA_Demo
Sleeping

App Files Community
GSridhar1982 commited on
Commit
cc7b5b0
·
verified ·
1 Parent(s): fdbd9dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import nltk
4
  nltk.download('punkt')
5
 
6
  def generate_answer(question):
7
- #model_name = "anukvma/bart-aiml-question-answer-v2"
8
- model_name = "GSridhar1982/AIML_QA_Llama31_FineTuned_UsingLora"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
10
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
11
  inputs = ["Answer this AIML Question: " + question]
 
4
  nltk.download('punkt')
5
 
6
  def generate_answer(question):
7
+ model_name = "anukvma/bart-aiml-question-answer-v2"
8
+ #model_name = "GSridhar1982/AIML_QA_Llama31_FineTuned_UsingLora"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
10
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
11
  inputs = ["Answer this AIML Question: " + question]