app.py

from flask import Flask, request, jsonify
from transformers import pipeline
import os

# Önemli: huggingface hub cache dizinini değiştir
os.environ['HF_HOME'] = '/tmp/huggingface'
os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers'

app = Flask(__name__)

classifier = pipeline("image-classification", model="ahmed792002/vit-plant-classification")

@app.route('/predict', methods=['POST'])
def predict():
    if 'image' not in request.files:
        return jsonify({'error': 'No image uploaded'}), 400

    image = request.files['image']
    predictions = classifier(image)
    return jsonify(predictions)