Gabriel commited on
Commit
9deafd0
1 Parent(s): e83839b
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from torch import tensor
4
  from torch.nn import functional as F
5
  import pandas as pd
6
 
7
- model = torch.load('model.pkl')
8
 
9
  def calc(X_test):
10
  # return model.predict(X_test)
 
4
  from torch.nn import functional as F
5
  import pandas as pd
6
 
7
+ # model = torch.load('model.pkl')
8
 
9
  def calc(X_test):
10
  # return model.predict(X_test)
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- fastai
 
 
1
+ fastai
2
+ torch