MohammadAliMKH commited on
Commit
712a870
β€’
1 Parent(s): ac12ce6

Update predict.py

Browse files

import efficient_model from model.py

Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  import torchvision
3
- from model import efficient_transformer
4
 
5
  FOOD101_CLASS_NAMES = [
6
  'apple_pie',
 
1
  import torch
2
  import torchvision
3
+ from model import efficient_transformer , efficient_model
4
 
5
  FOOD101_CLASS_NAMES = [
6
  'apple_pie',