Spaces:
Sleeping
Sleeping
dippatel1994
commited on
Commit
•
bb95cb0
1
Parent(s):
7b6e73c
Update app.py
Browse files
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)
|