Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 import torch
-from PIL import Image
 import open_clip
 
-def greet(image):
-    image = Image.fromarray(file_path)
+def predict(image):
     device = torch.device("cpu")
     model, _, preprocess = open_clip.create_model_and_transforms('ViT-L-14', pretrained='openai', device=device)
 
@@ -16,8 +14,13 @@ def greet(image):
             image_features /= image_features.norm(dim=-1, keepdim=True)
 
     embedding = image_features[0]
+    print(embedding)
 
-    return str(embedding)
+    model = joblib.load('model.pkl')
+    result = model.predict([embedding])
+    print(result)
 
-demo = gr.Interface(fn=greet, inputs=gr.inputs.Image(), outputs="text")
+    return "Map" if result == 1 else "No map"
+
+demo = gr.Interface(fn=predict, inputs=gr.Image(label="Input Image", type="pil"), outputs="text")
 demo.launch()