Spaces:
Sleeping
Sleeping
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)
|