Update app.py

app.py

@@ -3,9 +3,9 @@ import streamlit as st
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)
 
-# from datasets import load_dataset
+from datasets import load_dataset
 
-# dataset = load_dataset("soulhq-ai/insuranceQA-v2")
+dataset = load_dataset("soulhq-ai/insuranceQA-v2")
 
 import requests
 import os
@@ -21,9 +21,9 @@ def query(payload):
 	
 output = query({
 	"inputs": {
-    	"question": "What is my name?",
-    	"context": "My name is Clara and I live in Berkeley."
+    	"question": "What is my name?", 
+         "context": dataset
     }
 })
 
-st.write(output)
+st.write(output.answer)