Chintan-Shah commited on
Commit
6aa084f
·
verified ·
1 Parent(s): 950a747

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -8,5 +8,25 @@ sdk_version: 4.27.0
8
  app_file: app.py
9
  pinned: false
10
  ---
 
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  app_file: app.py
9
  pinned: false
10
  ---
11
+ # Gradio spaces on Huggingface for inferencing CustomResnet18 trained on CIFAR10 using Pytorch Lightning
12
 
13
+ ## Basic expectations
14
+ - Migrate Custom Resnet18 code from Pytorch to Pytorch Lightning first and then to Spaces such that:
15
+ - Migrate model on Lightning from Pytorch
16
+ - Use Gradio for deployment of Spaces app
17
+
18
+ - Spaces app has these features:
19
+ - Ask the user whether he/she wants to see GradCAM images
20
+ - How many
21
+ - From which layer
22
+ - Allow opacity change as well
23
+ - Ask whether he/she wants to view misclassified images
24
+ - How many
25
+ - Allow users to upload new images
26
+ - Provide 10 example images as well
27
+ - Ask how many top classes are to be shown (make sure the user cannot enter more than 10)
28
+
29
+
30
+ ## Reference to the repo used for training the Lightning model
31
+ - https://github.com/ChintanShahDS/ERAV2_Lit
32
+ - Follow this to try your own Resnet18 model with different hyperparameters and options on Pytorch Lightning