Update app.py
Browse files
app.py
CHANGED
|
@@ -71,13 +71,13 @@ ranks = [
|
|
| 71 |
|
| 72 |
def preprocess_data(data):
|
| 73 |
# Preprocess the data (replace this with your specific preprocessing steps)
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
|
| 78 |
-
|
| 79 |
|
| 80 |
-
return
|
| 81 |
|
| 82 |
# Define your prediction function
|
| 83 |
def make_prediction(rank,map,agent_picks):
|
|
|
|
| 71 |
|
| 72 |
def preprocess_data(data):
|
| 73 |
# Preprocess the data (replace this with your specific preprocessing steps)
|
| 74 |
+
data = ranks.index(data[0])
|
| 75 |
+
data[1] = maps.index(data[1])
|
| 76 |
+
data[2:7] = [agents.index(agent) for agent in data[2:7]]
|
| 77 |
|
| 78 |
+
data = torch.tensor(data, dtype = torch.float32)
|
| 79 |
|
| 80 |
+
return data
|
| 81 |
|
| 82 |
# Define your prediction function
|
| 83 |
def make_prediction(rank,map,agent_picks):
|