Upload 3 files

Dockerfile

@@ -0,0 +1,20 @@
+
+# Usa l'immagine base ufficiale di Python
+FROM python:3.8-slim
+
+# Installa le dipendenze
+COPY requirements.txt .
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+
+# Copia i pesi del modello
+COPY best.pt /app/best.pt
+
+# Copia lo script di inferenza
+COPY app.py /app/app.py
+
+# Imposta la directory di lavoro
+WORKDIR /app
+
+# Comando per eseguire lo script di inferenza
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,15 @@
+
+from ultralytics import YOLO
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+model = YOLO('best.pt')
+
+@app.route('/predict', methods=['POST'])
+def predict():
+    image = request.files['file'].read()
+    results = model(image)
+    return jsonify(results.pandas().xyxy[0].to_json(orient="records"))
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=8080)

requirements.txt

@@ -0,0 +1,6 @@
+
+transformers
+torch
+torchvision
+ultralytics
+flask