Hugging Face's logo

Spaces:
envision-ai
/
Custom-Object-Detection
Runtime error

App Files Community
Ubuntu commited on
Commit
020203d
•
1 Parent(s): 64dfa6f

fixes

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -0
  2. app.py +1 -0
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -5,6 +5,7 @@ from encoded_video import EncodedVideo, write_video
5
  import torch
6
  import numpy as np
7
  from torchvision.datasets import ImageFolder
 
8
  from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
9
  from pathlib import Path
10
  import pytorch_lightning as pl
 
5
  import torch
6
  import numpy as np
7
  from torchvision.datasets import ImageFolder
8
+ import transformers
9
  from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
10
  from pathlib import Path
11
  import pytorch_lightning as pl
app.py CHANGED
@@ -5,6 +5,7 @@ from encoded_video import EncodedVideo, write_video
5
  import torch
6
  import numpy as np
7
  from torchvision.datasets import ImageFolder
 
8
  from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
9
  from pathlib import Path
10
  import pytorch_lightning as pl
 
5
  import torch
6
  import numpy as np
7
  from torchvision.datasets import ImageFolder
8
+ import transformers
9
  from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
10
  from pathlib import Path
11
  import pytorch_lightning as pl