dippatel1994 commited on
Commit
bb95cb0
1 Parent(s): 7b6e73c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import streamlit as st
2
  import requests
3
  from transformers import pipeline, BertTokenizer
@@ -18,6 +21,7 @@ def generate_answers(questions, paper_link):
18
  answers = []
19
  for question in questions.split(","):
20
  inputs = tokenizer(question.strip(), paper_text, return_tensors="pt")
 
21
  answer = model(**inputs)
22
  answers.append(answer['answer'])
23
 
@@ -36,4 +40,4 @@ if st.button("Generate Answers"):
36
  with st.spinner("Generating answers..."):
37
  answers = generate_answers(questions, paper_link)
38
  st.success("Answers generated successfully!")
39
- st.text_area("Generated Answers", answers)
 
1
+ import os
2
+ os.environ["HF_HOME"] = "./transformers" # Set HF_HOME to a local directory to avoid TensorFlow dependency
3
+
4
  import streamlit as st
5
  import requests
6
  from transformers import pipeline, BertTokenizer
 
21
  answers = []
22
  for question in questions.split(","):
23
  inputs = tokenizer(question.strip(), paper_text, return_tensors="pt")
24
+ inputs = {k: v.tolist()[0] for k, v in inputs.items()} # Convert tensors to lists
25
  answer = model(**inputs)
26
  answers.append(answer['answer'])
27
 
 
40
  with st.spinner("Generating answers..."):
41
  answers = generate_answers(questions, paper_link)
42
  st.success("Answers generated successfully!")
43
+ st.text_area("Generated Answers", answers)