Update app.py

app.py

@@ -3,6 +3,7 @@ import time
 import pandas as pd
 import numpy as np
 import torch
+import requests
 from transformers import pipeline
 
 st.set_page_config(page_title="Samuel Portfolio", page_icon="📈")
@@ -87,14 +88,18 @@ if choice == "Plotting":
     st.button("Re-run")
 
 if choice == "Contact":
-    st.title("You can contact me via:")
-
-    torch.manual_seed(0)
-    generator = pipeline('text-generation', model = 'gpt2')
-    prompt = "Hello, I'm a language model"
+    st.title("You can contact me via:")    
+    
+    API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
+    headers = {"Authorization": "Bearer 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"}
     
-    generator(prompt, max_length = 30)
-    st.write(generator)
+    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 ",
+    })
 
 if choice == "Attached files": 
     st.title("Download final project report here")