Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
adding model predictions
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ class Engine(object):
|
|
72 |
# Get the image filenames and their coordinates
|
73 |
self.images = df['id'].tolist()[:]
|
74 |
self.coordinates = df[['longitude', 'latitude']].values.tolist()[:]
|
75 |
-
self.admins = df[['city', '
|
76 |
self.preds = df[['pred_longitude', 'pred_latitude']].values.tolist()[:]
|
77 |
|
78 |
def isfinal(self):
|
|
|
72 |
# Get the image filenames and their coordinates
|
73 |
self.images = df['id'].tolist()[:]
|
74 |
self.coordinates = df[['longitude', 'latitude']].values.tolist()[:]
|
75 |
+
self.admins = df[['city', 'sub-region', 'region', 'country']].values.tolist()[:]
|
76 |
self.preds = df[['pred_longitude', 'pred_latitude']].values.tolist()[:]
|
77 |
|
78 |
def isfinal(self):
|