Create README.md

README.md

@@ -0,0 +1,38 @@
+```
+Copied
+import json
+import requests
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+API_URL = "https://api-inference.huggingface.co/models/sillon/DialoGPT-small-HospitalBot"
+def query(payload):
+    data = json.dumps(payload)
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+    return json.loads(response.content.decode("utf-8"))
+data = query(
+    {
+        "inputs": {
+            "past_user_inputs": ["Which movie is the best ?"],
+            "generated_responses": ["It's Die Hard for sure."],
+            "text": "Can you explain why ?",
+        },
+    }
+)
+# Response
+self.assertEqual(
+    data,
+    {
+        "generated_text": "It's the best movie ever.",
+        "conversation": {
+            "past_user_inputs": [
+                "Which movie is the best ?",
+                "Can you explain why ?",
+            ],
+            "generated_responses": [
+                "It's Die Hard for sure.",
+                "It's the best movie ever.",
+            ],
+        },
+        "warnings": ["Setting `pad_token_id` to `eos_token_id`:50256 for open-end generation."],
+    },
+)
+```