grover101 commited on
Commit
90359a4
·
verified ·
1 Parent(s): 8cc6518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,7 +4,9 @@ import gradio as gr
4
  def old_or_not(x):
5
  return x[0].isupper()
6
 
7
- learn = load_model('model.pkl')
 
 
8
 
9
  categories = ['New_car', 'Old_car']
10
 
 
4
  def old_or_not(x):
5
  return x[0].isupper()
6
 
7
+
8
+ learn = load_model('model.pkl', arch='resnet18', pretrained=True)
9
+
10
 
11
  categories = ['New_car', 'Old_car']
12