Hugging Face's logo

Spaces:
chcomet
/
cholec80-position-encoder
Runtime error

App Files Community
Hui commited on
Commit
8b98c60
•
1 Parent(s): 4e6b503

requirements

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. app.py +1 -3
  3. models.py +0 -1
  4. requirements.txt +2 -0
.gitattributes CHANGED
@@ -2,6 +2,7 @@
2
  *.arrow filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  *.bz2 filter=lfs diff=lfs merge=lfs -text
 
5
  *.ftz filter=lfs diff=lfs merge=lfs -text
6
  *.gz filter=lfs diff=lfs merge=lfs -text
7
  *.h5 filter=lfs diff=lfs merge=lfs -text
2
  *.arrow filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
  *.ftz filter=lfs diff=lfs merge=lfs -text
7
  *.gz filter=lfs diff=lfs merge=lfs -text
8
  *.h5 filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -1,8 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from PIL import Image
4
- from torch import nn
5
- from torchvision import models, transforms
6
 
7
  # parameters
8
  from models import Cholec80Model
1
  import gradio as gr
2
  import torch
3
+ from torchvision import transforms
 
 
4
 
5
  # parameters
6
  from models import Cholec80Model
models.py CHANGED
@@ -1,4 +1,3 @@
1
- from dataclasses import dataclass
2
 
3
  import torch
4
  import torchvision.models as models
 
1
 
2
  import torch
3
  import torchvision.models as models
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
1
+ torch
2
+ torchvision