samuel2424 commited on
Commit
082a8be
·
1 Parent(s): b112629

Adding app file

Browse files
Files changed (2) hide show
  1. README.md +7 -7
  2. model.py +2 -1
README.md CHANGED
@@ -1,13 +1,13 @@
1
  ---
2
- title: Classifier
3
- emoji: 🌖
4
- colorFrom: green
5
- colorTo: blue
6
- sdk: streamlit
7
- sdk_version: 1.29.0
8
  app_file: app.py
9
  pinned: false
10
  license: mit
11
  ---
12
 
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: Weather Image Classifier
3
+ emoji:
4
+ colorFrom: indigo
5
+ colorTo: green
6
+ sdk: gradio
7
+ sdk_version: 3.16.2
8
  app_file: app.py
9
  pinned: false
10
  license: mit
11
  ---
12
 
13
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
model.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  import torchvision
3
  from torch import nn
@@ -21,4 +22,4 @@ def create_effnetb2_model(num_classes: int):
21
  nn.Linear(in_features=1408, out_features=num_classes)
22
  )
23
 
24
- return model, transforms
 
1
+
2
  import torch
3
  import torchvision
4
  from torch import nn
 
22
  nn.Linear(in_features=1408, out_features=num_classes)
23
  )
24
 
25
+ return model, transforms