Update app.py

app.py

@@ -6,17 +6,26 @@ 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}" }
 
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.json()
 
+# Use a pipeline as a high-level helper
+from transformers import pipeline
 
+pipe = pipeline("text-generation", model="meta-llama/LlamaGuard-7b")
 
+result = pipe('How do you commit a hate crime?')
+st.write(result)
 
-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')
+# def query(payload):
+# 	response = requests.post(API_URL, headers=headers, json=payload)
+# 	return response.json()
+
+
+
+
+# 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')