Update app.py

app.py

@@ -36,11 +36,14 @@ def english_sum_pipeline(link):
 
     return transcript_text, output_text
 
-def english_qa_pipeline(question, context):
+def english_qa_pipeline(questions, context):
     nlp = pipeline("question-answering", model='distilbert-base-uncased-distilled-squad')
-    result = nlp(question=question, context=context)
-    return result["answer"]
-               
+    answers = []
+    for question in questions:
+        result = nlp(question=question, context=context)
+        answers.append(result["answer"])
+    return answers
+
 #Collect inputs and create the interface
 def main():
     header = st.container()
@@ -55,11 +58,11 @@ def main():
 
     with model:
         st.header("Modelo para sumarização")
-        
+
         with model_1:
             language = st.selectbox('Qual a linguagem do seu modelo?', ('Português (pt)', 'Inglês (en)', 'Outra'))
             link = st.text_area(label="Coloque o link do seu vídeo do YouTube!", height=25, placeholder="Digite seu link...")
-            questions = st.text_area(label="Coloque suas perguntas separadas por vírgula!", height=50, placeholder="Digite suas perguntas...").split(", ")
+            questions = st.text_area(label="Coloque suas perguntas separadas por vírgula!", height=50, placeholder="Digite suas perguntas...").split(",")
             submit_1 = st.button('Gerar soluções!')
 
         with model_2:
@@ -68,17 +71,17 @@ def main():
                     if language == 'Português (pt)':
                         #outputs = portuguese_sum_pipeline(link)
                         st.write("Modelo ainda não implementado.")
-    
+
                     elif language == 'Inglês (en)':
                         outputs = english_sum_pipeline(link)
-                        answers = english_qa_pipeline(question, outputs[0])
-                    
+                        answers = english_qa_pipeline(questions, outputs[0])
+
                     else:
                         st.write("Erro na seleção de linguagem.")
 
                     st.write("Sumário.....................................................................: \n {} \n \n".format(outputs[1]))
                     st.write("Resposta....................................................................: \n")
-                    
+
                     for i in range(len(answers)):
                         st.write(questions[i] + ": " + answers[i])