rhazeljay commited on
Commit
bff2b98
·
verified ·
1 Parent(s): b86d733

update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -17
app.py CHANGED
@@ -14,23 +14,6 @@ configure(api_key="AIzaSyBawh403z5cyyQzFhQo14y7oUQw6nr8mIg")
14
  model = GenerativeModel("gemini-2.0-flash")
15
 
16
  # Load DeepLabV3 model for garment segmentation
17
- def load_segmentation_model():
18
- model = models.segmentation.deeplabv3_resnet101(pretrained=True)
19
- model.eval()
20
- return model
21
-
22
- # Load DeepLabV3 model for garment segmentation
23
- def load_segmentation_model():
24
- model = models.segmentation.deeplabv3_resnet101(pretrained=True)
25
- model.eval()
26
- return model
27
-
28
- import torch
29
- import numpy as np
30
- import torchvision.transforms as transforms
31
- import torchvision.models as models
32
- from PIL import Image
33
-
34
  def load_segmentation_model():
35
  model = models.segmentation.deeplabv3_resnet101(pretrained=True)
36
  model.eval()
 
14
  model = GenerativeModel("gemini-2.0-flash")
15
 
16
  # Load DeepLabV3 model for garment segmentation
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  def load_segmentation_model():
18
  model = models.segmentation.deeplabv3_resnet101(pretrained=True)
19
  model.eval()