aliabd HF staff commited on
Commit
8a1d2d9
β€’
1 Parent(s): 3a247f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -3,7 +3,11 @@ from pathlib import Path
3
  import torch
4
  import gradio as gr
5
  from torch import nn
 
6
 
 
 
 
7
 
8
  LABELS = Path('class_names.txt').read_text().splitlines()
9
 
3
  import torch
4
  import gradio as gr
5
  from torch import nn
6
+ import gdown
7
 
8
+ url = 'https://drive.google.com/uc?id=1dsk2JNZLRDjC-0J4wIQX_FcVurPaXaAZ'
9
+ output = 'pytorch_model.bin'
10
+ gdown.download(url, output, quiet=False)
11
 
12
  LABELS = Path('class_names.txt').read_text().splitlines()
13