Update app.py

app.py

@@ -1,16 +1,18 @@
 from flask import Flask, request, jsonify
-from transformers import AutoModelForImageClassification, AutoProcessor
+from flask_cors import CORS
+from transformers import Dinov2ForImageClassification, AutoProcessor # Import the correct model class.
 from PIL import Image
 import io
-import fitz  # PyMuPDF
-from flask_cors import CORS
+import fitz
+import torch
 
 app = Flask(__name__)
-CORS(app)  # Allow cross-origin requests
+CORS(app)
 
 # Load model and processor
 model_name = "AsmaaElnagger/Diabetic_RetinoPathy_detection"
-model = AutoModelForImageClassification.from_pretrained(model_name)
+# model = AutoModelForImageClassification.from_pretrained(model_name) # REMOVE THIS LINE
+model = Dinov2ForImageClassification.from_pretrained(model_name)  # USE THE CORRECT MODEL CLASS
 processor = AutoProcessor.from_pretrained(model_name)
 
 def pdf_to_images_pymupdf(pdf_data):
@@ -20,7 +22,7 @@ def pdf_to_images_pymupdf(pdf_data):
         for page_num in range(pdf_document.page_count):
             page = pdf_document.load_page(page_num)
             pix = page.get_pixmap()
-            img_data = pix.tobytes("jpeg")  # Or "png"
+            img_data = pix.tobytes("jpeg")
             images.append(img_data)
         return images
     except Exception as e:
@@ -71,4 +73,4 @@ def classify_file():
         return jsonify({'error': f'An error occurred: {e}'}), 500
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860, debug=True)  # Make it accessible from any network interface
+    app.run(host="0.0.0.0", port=7860, debug=True)  # Make it accessible from any network interface