Add application file

Author: SalimM21

Dockerfile

@@ -1,16 +1,31 @@
-FROM python:3.11-slim
+FROM python:3.10-slim
 
-ENV PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1
+# Installer dépendances système nécessaires pour torch et transformers
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    git \
+    && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
+# Copier le fichier requirements.txt
 COPY requirements.txt .
+
+# Upgrade pip
+RUN pip install --upgrade pip
+
+# Installer les dépendances Python, incluant torch CPU
+RUN pip install --no-cache-dir torch==2.2.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html
+RUN pip install --no-cache-dir "numpy<2"
 RUN pip install --no-cache-dir -r requirements.txt
 
-COPY app ./app
+# Copier le reste du projet
+COPY . .
 COPY static ./static
 
+# Exposer le port 7860 (obligatoire pour HF Spaces)
 EXPOSE 7860
-# Hugging Face Spaces fournit $PORT
-CMD uvicorn app.main:app --host 0.0.0.0 --port ${PORT:-7860}
+
+# Lancer FastAPI via Uvicorn (forme JSON recommandée)
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+from transformers import pipeline
+import os
+
+# Charger le modèle (identique à celui de ton API)
+MODEL_NAME = os.getenv("HF_MODEL", "distilbert-base-uncased-finetuned-sst-2-english")
+sentiment_pipe = pipeline("sentiment-analysis", model=MODEL_NAME)
+
+# Fonction de prédiction
+def predict_sentiment(text):
+    result = sentiment_pipe(text)[0]
+    label = "positive" if result["label"].upper().startswith("POS") else "negative"
+    confidence = float(result["score"])
+    return f"Label: {label}, Confidence: {confidence:.4f}"
+
+# Interface Gradio
+iface = gr.Interface(
+    fn=predict_sentiment,
+    inputs=gr.Textbox(lines=3, placeholder="Entrez votre texte ici..."),
+    outputs="text",
+    title="Sentiment Analysis Demo",
+    description=f"Modèle: {MODEL_NAME}"
+)
+
+# Lancer l'interface en local
+if __name__ == "__main__":
+    iface.launch(share=True)
+