update spaces

.gitattributes

@@ -38,3 +38,4 @@ model/190703-214543/loss.log filter=lfs diff=lfs merge=lfs -text
 model/190703-214543/model_95.ckpt.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
 model/190703-214543/model_95.ckpt.index filter=lfs diff=lfs merge=lfs -text
 model/190703-214543/model_95.ckpt.meta filter=lfs diff=lfs merge=lfs -text
+tests/test.mseed filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,4 +1,13 @@
 ---
+title: Test Model
+emoji: 🐨
+colorFrom: pink
+colorTo: blue
+sdk: gradio
+sdk_version: 3.15.0
+app_file: app.py
+pinned: false
+license: mit
 tags:
 - text-classification
 library_name: generic

pipeline.py

@@ -32,7 +32,7 @@ class PreTrainedPipeline():
         self.sess = sess
         self.model = model
 
-    def __call__(self, inputs: str) -> List[List[Dict[str, float]]]:
+    def __call__(self, inputs: str, **kwargs) -> List[List[Dict[str, float]]]:
         """
         Args:
             inputs (:obj:`str`):

requirements.txt

@@ -1 +1,4 @@
-tensorflow
+tensorflow
+numpy
+obspy
+pandas

tests/test_api.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -12,7 +12,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
@@ -31,14 +31,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 29,
+   "execution_count": 15,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "waveform.write(\"test.mseed\", format=\"MSEED\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[[{'phase_index': 470, 'phase_score': 0.954, 'phase_type': 'P'}, {'phase_index': 570, 'phase_score': 0.839, 'phase_type': 'S'}]]\n"
+      "[[{'label': 'I like you. I love you', 'score': 0.2}, {'args': []}, {}]]\n"
      ]
     }
    ],
@@ -47,8 +56,8 @@
     "import numpy as np\n",
     "import json\n",
     "\n",
-    "API_URL = \"https://api-inference.huggingface.co/models/zhuwq/PhaseNet\"\n",
-    "# API_URL = \"https://api-inference.huggingface.co/models/zhuwq/test-model\"\n",
+    "# API_URL = \"https://api-inference.huggingface.co/models/zhuwq/PhaseNet\"\n",
+    "API_URL = \"https://api-inference.huggingface.co/models/zhuwq/test-model\"\n",
     "headers = {\"Authorization\": \"Bearer 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\"}\n",
     "\n",
     "def query(payload):\n",
@@ -57,11 +66,13 @@
     "    # return json.loads(response.content.decode(\"utf-8\"))\n",
     "\n",
     "# array = np.random.rand(10, 3).tolist()\n",
-    "inputs = json.dumps(array.tolist())\n",
+    "# inputs = json.dumps(array.tolist())\n",
     "data = {\n",
-    "\t# \"inputs\": \"I like you. I love you\",\n",
-    "    \"inputs\": inputs,\n",
-    "    \"options\":{\"wait_for_model\": True},\n",
+    "\t\"inputs\": \"I like you. I love you\",\n",
+    "    \"arg\": 1,\n",
+    "    # \"inputs\": inputs,\n",
+    "    \"options\": {\"wait_for_model\": True},\n",
+    "    \"parameters\": {\"max_length\": 500}\n",
     "}\n",
     "\n",
     "output = query(data)\n",