set-user-agent

app.py

@@ -51,7 +51,10 @@ def infer(
         "stop": stop,
     }
     print(f"send: {datetime.now()}")
-    response = requests.get("https://staging.together.xyz/api/inference", params=my_post_dict).json()
+    headers = {
+        'User-Agent': 'GPT-JT HuggingFace Space'
+    }
+    response = requests.get("https://staging.together.xyz/api/inference", params=my_post_dict, headers=headers).json()
     generated_text = response['output']['choices'][0]['text']
     print(f"recv: {datetime.now()}")
     
@@ -277,4 +280,4 @@ def main():
             generated_area.markdown("<b>" + to_md(prompt) + "</b><mark style='background-color: #cbeacd'>" + to_md(report_text)+"</mark>", unsafe_allow_html=True)
         
 if __name__ == '__main__':
-    main()
+    main()