Hugging Face's logo

Spaces:
eternalBlissard
/
FoodVision101
Runtime error

App Files Community
1
eternalBlissard commited on
Commit
71c2fd4
·
verified ·
1 Parent(s): 294143b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -7,7 +7,7 @@ from ViT import ViT
7
  import spaces
8
 
9
 
10
- @spaces.GPU(duration=5)
11
  def getViT(seed,classNames,DEVICE):
12
  setAllSeeds(seed)
13
  ViTModel = ViT(3,768,16,224,3072,12,0.1,12,len(classNames)).to(DEVICE)
 
7
  import spaces
8
 
9
 
10
+ # @spaces.GPU(duration=5)
11
  def getViT(seed,classNames,DEVICE):
12
  setAllSeeds(seed)
13
  ViTModel = ViT(3,768,16,224,3072,12,0.1,12,len(classNames)).to(DEVICE)