Update app.py

app.py

@@ -53,15 +53,10 @@ def image_resize(data):
 if file is not None:
     df = pd.read_csv(file)  
     # Get first row 
-    row = df.iloc[0]  
-    
-    # Load image 
-    # image = np.array(Image.open(row[0])) 
-    # Reshape  
-    img_reshaped = image_resize(row)
+    img_reshaped = image_resize(df)
     
     # Get prediction
-    pred = model.predict(img_reshaped)
+    pred = model.predict(img_reshaped[0])
     label = np.argmax(pred)
 
     label_map = {4: ('nv', ' melanocytic nevi'),