Create app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+MODEL_NAME = "ton-username/ton-modele"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt", truncation=True)
+    outputs = model(**inputs)
+    probs = torch.nn.functional.softmax(outputs.logits, dim=1)
+    predicted_class = torch.argmax(probs).item()
+    confidence = probs[0][predicted_class].item()
+    
+    return f"Classe: {predicted_class} (confiance: {confidence:.2f})"
+
+interface = gr.Interface(
+    fn=predict,
+    inputs="text",
+    outputs="text",
+    title="Mon modèle de classification",
+    description="Entrez un texte pour prédiction."
+)
+
+interface.launch()