Piraloco commited on
Commit
5daefae
·
1 Parent(s): 29bb5f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- processed_data = ranks.index(processed_data[0])
75
- processed_data[1] = maps.index(processed_data[1])
76
- processed_data[2:7] = [agents.index(agent) for agent in processed_data[2:7]]
77
 
78
- inputs = torch.tensor(processed_data, dtype = torch.float32)
79
 
80
- return processed_data
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):