fix JIT 3

app.py

@@ -56,8 +56,7 @@ def process_image(image):
         outputs = traced_model(**inputs)
 
     # Normalize the features before search, whatever that means
-    embeddings = outputs.last_hidden_state
-    embeddings = embeddings.mean(dim=1)
+    embeddings = outputs[0].mean(dim=1)
     vector = embeddings.detach().cpu().numpy()
     vector = np.float32(vector)
     faiss.normalize_L2(vector)