Switched to Facebook model

app.py

@@ -14,7 +14,7 @@ file_path = "Adrega_P.I._User_Manual.docx"
 #pipe = pipeline("question-answering")
 #pipe = pipeline("text-generation")
 #pipe = pipeline("question-answering", model="deepset/roberta-base-squad2")
-pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
+pipe = pipeline("summarization", model="facebook/bart-large-cnn")
 
 st.title("Adrega AI Help")
 #dataset = load_dataset("andreska/adregadocs", split="test")
@@ -28,9 +28,11 @@ if st.button("Submit"):
         #text_inputs = f"Context: {context}\nQuestion: {user_input}\nAnswer:"
         #result = pipe(text_inputs, max_length=200, num_return_sequences=1)[0]['generated_text']
         #answer = result.split("Answer:")[1].strip()
-        answer = pipe(question=user_input, context=context)
+        #answer = pipe(question=user_input, context=context)
+        answer = pipe(context, max_length=512, min_length=100, do_sample=False) 
+        st.write(f"Adrega AI: {answer[0]['summary_text']}")
         
-        st.write(f"Adrega AI: {answer}")
+        #st.write(f"Adrega AI: {answer}")
     else: 
         st.write("Please enter a question.")