islasher commited on
Commit
af8ae33
1 Parent(s): ad48346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -2,12 +2,9 @@
2
  #app.py:
3
  # from huggingface_hub import from_pretrained_fastai
4
  import gradio as gr
5
- from fastcore.xtras import Path
6
- from fastai.callback.hook import summary
7
- from fastai.callback.progress import ProgressCallback
8
- from fastai.callback.schedule import lr_find, fit_flat_cos
9
  from fastai.data.block import DataBlock
10
- from fastai.data.external import untar_data, URLs
11
  from fastai.data.transforms import get_image_files, FuncSplitter, Normalize
12
  from fastai.layers import Mish
13
  from fastai.losses import BaseLoss
@@ -20,13 +17,13 @@ from fastai.vision.learner import unet_learner
20
  from PIL import Image
21
  import numpy as np
22
  from torch import nn
23
- from torchvision.models.resnet import resnet34
24
  import torch
25
  import torch.nn.functional as F
26
 
27
 
28
  # # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
29
  repo_id = "islasher/segm-grapes"
 
30
 
31
 
32
  # # Definimos una función que se encarga de llevar a cabo las predicciones
 
2
  #app.py:
3
  # from huggingface_hub import from_pretrained_fastai
4
  import gradio as gr
5
+
 
 
 
6
  from fastai.data.block import DataBlock
7
+
8
  from fastai.data.transforms import get_image_files, FuncSplitter, Normalize
9
  from fastai.layers import Mish
10
  from fastai.losses import BaseLoss
 
17
  from PIL import Image
18
  import numpy as np
19
  from torch import nn
 
20
  import torch
21
  import torch.nn.functional as F
22
 
23
 
24
  # # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
25
  repo_id = "islasher/segm-grapes"
26
+ repo_id='segm-grapes'
27
 
28
 
29
  # # Definimos una función que se encarga de llevar a cabo las predicciones