Update app.py

app.py

@@ -3,22 +3,20 @@ import streamlit as st
 
 import os
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/meta-llama/LlamaGuard-7b"
+headers = {"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}" }
 
-st.button("Reset", type="primary")
-if st.button('Say hello'):
-    st.write('Why hello there')
-else:
-    st.write('Goodbye')
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.json()
 
-    
 
-# API_URL = "https://api-inference.huggingface.co/models/meta-llama/LlamaGuard-7b"
-# headers = {"Authorization": "Bearer {HUGGINGFACEHUB_API_TOKEN}"}
 
-# def query(payload):
-# 	response = requests.post(API_URL, headers=headers, json=payload)
-# 	return response.json()
 
-# output = query({
-# 	"inputs": "Can you please let us know more details about your ",
-# })
+st.button("Reset", type="primary")
+if st.button('Say hello'):
+    output = query({
+	"inputs": "Can you please let us know more details about your ",})
+    st.write(output)
+else:
+    st.write('Goodbye')