sadjava commited on
Commit
caa33e3
1 Parent(s): 4b1ad8f

fixed errors

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +9 -0
README.md CHANGED
@@ -2,7 +2,7 @@
2
  title: Dog Localization
3
  emoji: 🐶
4
  colorFrom: yellow
5
- colorTo: brown
6
  sdk: gradio
7
  sdk_version: 3.33.1
8
  app_file: app.py
 
2
  title: Dog Localization
3
  emoji: 🐶
4
  colorFrom: yellow
5
+ colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.33.1
8
  app_file: app.py
app.py CHANGED
@@ -4,6 +4,15 @@
4
  __all__ = ['device', 'model', 'MEAN', 'STD', 'transform', 'image', 'label', 'examples', 'intf', 'to_img', 'draw_image_with_bbox',
5
  'localize_dog']
6
 
 
 
 
 
 
 
 
 
 
7
  # %% ../app.ipynb 3
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
 
4
  __all__ = ['device', 'model', 'MEAN', 'STD', 'transform', 'image', 'label', 'examples', 'intf', 'to_img', 'draw_image_with_bbox',
5
  'localize_dog']
6
 
7
+ # %% ../app.ipynb 2
8
+ from model import Model
9
+ import torch
10
+ from torchvision import transforms
11
+ import gradio as gr
12
+ import numpy as np
13
+ from PIL import Image, ImageDraw
14
+ import cv2
15
+
16
  # %% ../app.ipynb 3
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
18