add tokenizer

app.py

@@ -10,14 +10,17 @@
 # write a gradio interface for tabibu-ai/mental-health-chatbot in huggingfacehub
 
 # Path: app.py
+import pickle
 import gradio as gr
 
 # install transformers and torch in requirements.txt
 
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-tokenizer = AutoTokenizer.from_pretrained("tabibu-ai/mental-health-chatbot")
-model = AutoModelForSequenceClassification.from_pretrained("tabibu-ai/mental-health-chatbot")
+# tokenizer = AutoTokenizer.from_pretrained("tabibu-ai/mental-health-chatbot")
+
+tokenizer = AutoTokenizer.from_pretrained("rabiaqayyum/autotrain-mental-health-analysis-752423172")
+model = pickle.load(open("model.pkl", "rb"))
 
 def classify_text(inp):
     input_ids = tokenizer.encode(inp, return_tensors='pt')

model.ipynb

@@ -2,19 +2,10 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "ace57031",
    "metadata": {},
    "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "2023-02-21 17:41:49.330107: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory\n",
-      "2023-02-21 17:41:49.330992: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory\n",
-      "2023-02-21 17:41:49.331010: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.\n"
-     ]
-    },
     {
      "data": {
       "text/html": [
@@ -92,7 +83,7 @@
        "4  When healing from mental illness, early identi...  "
       ]
      },
-     "execution_count": 1,
+     "execution_count": 2,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -141,7 +132,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 3,
    "id": "8f51e39d",
    "metadata": {},
    "outputs": [
@@ -162,7 +153,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 4,
    "id": "1d697a39",
    "metadata": {},
    "outputs": [
@@ -190,7 +181,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 5,
    "id": "c5dde0e4",
    "metadata": {},
    "outputs": [
@@ -198,7 +189,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Accuracy: 0.03333333333333333\n"
+      "Accuracy: 0.06666666666666667\n"
      ]
     }
    ],
@@ -247,20 +238,28 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 8,
    "id": "14406312",
    "metadata": {},
    "outputs": [],
    "source": [
-    "new_question =  \"What are the symptoms of depression?\"\n"
+    "new_question = input(\"Ask me anything : \")\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "id": "6b9198db",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Prediction: ['\"Different kinds of therapy are more effective based on the nature of the mental health condition and/or symptoms and the person who has them (for example']\n"
+     ]
+    }
+   ],
    "source": [
     "new_question_vector = vectorizer.transform([new_question])\n",
     "prediction = model.predict(new_question_vector)\n",