Update app.py

app.py

@@ -12,7 +12,7 @@ import os
 
 API_KEY = os.environ.get("API_KEY")
 
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-3.2-3B"
+API_URL = "API_URL = "https://api-inference.huggingface.co/models/google-bert/bert-large-uncased-whole-word-masking-finetuned-squad"
 headers = {"Authorization": "Bearer " + API_KEY}
 
 def query(payload):
@@ -20,7 +20,10 @@ def query(payload):
 	return response.json()
 	
 output = query({
-	"inputs": "Can you please let us know more details about your ",
+	"inputs": {
+    	"question": "What is my name?",
+    	"context": "My name is Clara and I live in Berkeley."
+    }
 })
 
 st.write(output)