Update app.py

app.py

@@ -1,34 +1,38 @@
-from flask import Flask, request, jsonify
-from transformers import pipeline
-from PIL import Image
-import io
-import torch
-
-app = Flask(__name__)
-
-classifier = pipeline("image-classification", model="umutbozdag/plant-identity")
-
-@app.route("/predict", methods=["POST"])
-def predict():
-    if "file" not in request.files:
-        return jsonify({"error": "No file uploaded"}), 400
-
-    file = request.files["file"]
-    image_bytes = file.read()
-    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-
-    try:
-        results = classifier(image)
-        top_result = results[0]
-        label = top_result["label"]
-        score = round(top_result["score"] * 100, 2)
-        return jsonify({"label": label, "confidence": f"{score}%"})
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
-@app.route("/", methods=["GET"])
-def home():
-    return "Bitki Tanıma API - Lumi 🌿"
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+import os
+os.environ['TRANSFORMERS_CACHE'] = '/data'
+
+
+from flask import Flask, request, jsonify
+from transformers import pipeline
+from PIL import Image
+import io
+import torch
+
+app = Flask(__name__)
+
+classifier = pipeline("image-classification", model="umutbozdag/plant-identity")
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    if "file" not in request.files:
+        return jsonify({"error": "No file uploaded"}), 400
+
+    file = request.files["file"]
+    image_bytes = file.read()
+    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+
+    try:
+        results = classifier(image)
+        top_result = results[0]
+        label = top_result["label"]
+        score = round(top_result["score"] * 100, 2)
+        return jsonify({"label": label, "confidence": f"{score}%"})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/", methods=["GET"])
+def home():
+    return "Bitki Tanıma API - Lumi 🌿"
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)