included the models directory

.gitignore

@@ -9,8 +9,5 @@ __pycache__/
 # DataSet
 data/
 
-# Exclude the models directory
-models/
-
 # Exclude the notebooks directory
 notebooks/

app/models/getModel.py

@@ -0,0 +1,9 @@
+from transformers import pipeline
+
+pipe = pipeline("image-classification", "anurag629/swin-tiny-patch4-window7-224-finetuned-eurosat")
+
+def getPrediction(image):
+    '''
+    Returns the prediction of the image.
+    '''
+    return pipe(image)