Hugging Face's logo

Spaces:
adiren7
/
FoodVisionVIT
Runtime error

App Files Community
adiren7 commited on
Commit
da25977
·
1 Parent(s): 40b6d6e

2nd commit

Browse files
Files changed (3) hide show
  1. __pycache__/model.cpython-39.pyc +0 -0
  2. app.py +1 -1
  3. requirements.txt +1 -0
__pycache__/model.cpython-39.pyc ADDED
Binary file (698 Bytes). View file
 
app.py CHANGED
@@ -10,7 +10,7 @@ class_names = ["pizza", "steak", "sushi"]
10
 
11
  vit , vit_transforms = create_vit_model()
12
 
13
- vit.load_state_dict(torch.laod(f="09_pretrained_vit_feature_extractor_pizza_steak_sushi_20_percent.pth",
14
  map_location=torch.device("cpu")))
15
 
16
  def predict(img):
 
10
 
11
  vit , vit_transforms = create_vit_model()
12
 
13
+ vit.load_state_dict(torch.load(f="09_pretrained_vit_feature_extractor_pizza_steak_sushi_20_percent.pth",
14
  map_location=torch.device("cpu")))
15
 
16
  def predict(img):
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch==1.12.0
2
  torchvision==0.13.0
3
  gradio==3.1.4
 
 
1
  torch==1.12.0
2
  torchvision==0.13.0
3
  gradio==3.1.4
4
+ httpx==0.24.1