Update

README.md

@@ -1,8 +1,8 @@
 ---
 title: Disaster
-emoji: 🦀
-colorFrom: indigo
-colorTo: blue
+emoji: 💥
+colorFrom: red
+colorTo: orange
 sdk: gradio
 sdk_version: 3.50.2
 app_file: app.py

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+model_name = "Daniel-Sousa/outputs"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+mapping = {
+    0: 'It is not a disaster',
+    1: 'Disaster',
+}
+
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt")
+
+    outputs = model(**inputs)
+    predictions = outputs.logits
+
+    return mapping[round(predictions.item())]
+
+iface = gr.Interface(
+    fn=predict,
+    inputs="text",
+    outputs="text",
+    layout="vertical",
+    title="Classificador de Tweets",
+    description="Esse modelo é capaz de realizar uma análise de frases em Inglês do Tweet e diser se ela está refenciado um desastre ou não"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,6 @@
+pandas==1.4.2
+Transformers==4.33.0
+Pytorch==2.0.0
+Datasets==2.1.0
+Tokenizers==0.13.3
+torch