Rename app.txt to app.py

app.txt → app.py

@@ -8,7 +8,7 @@ import requests
 
 from share_btn import community_icon_html, loading_icon_html, share_js
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
+word_list_dataset = load_dataset("shawarmas/Censored-Words", data_files="test.txt", use_auth_token=True)
 word_list = word_list_dataset["train"]['text']
 
 is_gpu_busy = False
@@ -20,8 +20,10 @@ def infer(prompt, negative, scale):
         
     images = []
     url = os.getenv('JAX_BACKEND_URL')
+    HF_TOKEN = os.environ.get("HF_READ_TOKEN")
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
-    images_request = requests.post(url, json = payload)
+    images_request = requests.post(url, headers=headers, json = payload)
     for image in images_request.json()["images"]:
         image_b64 = (f"data:image/jpeg;base64,{image}")
         images.append(image_b64)