Update app.py

fixed to do tech data classification

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("FuriouslyAsleep/unhappyZebra100")
+
+model = AutoModelForSequenceClassification.from_pretrained("FuriouslyAsleep/unhappyZebra100")
 
 def greet(name):
-    return "Hello " + name + "!!"
+
+    inputs = tokenizer(name, return_tensors="pt")
+    outputs = model(**inputs)
+    
+    predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    print(predictions)    
+#    return "Hello " + name + "!!"
+    return "Probabilities are listed here (False prob, then True prob): ) + predictions
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface.launch()
+