new

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
-import tensorflow as tf
+import os
+os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification, TextClassificationPipeline
 
 tokenizer = AutoTokenizer.from_pretrained("dipesh/Intent-Classification-Bert-Base-Cased")
@@ -15,7 +17,6 @@ def predict(input_text):
     return ans[0]['label']
 
 
-predict("hello")
-
-iface = gr.Interface(fn=predict, inputs="text", outputs="text")
+iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="Intent Classifier",
+                     description="Classifier")
 iface.launch()