fohy24 commited on
Commit
164d735
·
1 Parent(s): 2ded624

changed out examples for new morphs

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -49,20 +49,20 @@ model_path = hf_hub_download(repo_id="samfhy/morphmarket_model", filename="model
49
  checkpoint = torch.load(model_path, map_location=device)
50
 
51
  new_layers = nn.Sequential(
52
- nn.LazyLinear(2048),
53
- nn.BatchNorm1d(2048),
54
- nn.ReLU(),
55
- nn.Dropout(0.5),
56
- nn.LazyLinear(num_labels)
57
- )
58
 
59
  IMAGE_SIZE = 480
60
  transform = v2.Compose([
61
- v2.ToImage(),
62
- v2.Resize((IMAGE_SIZE, IMAGE_SIZE)),
63
- v2.ToDtype(torch.float32),
64
- v2.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
65
- ])
66
 
67
  efficientnet = models.efficientnet_v2_l(weights='EfficientNet_V2_L_Weights.DEFAULT')
68
  efficientnet.classifier = new_layers
@@ -87,7 +87,7 @@ gr.Interface(fn=predict,
87
  inputs=[gr.Image(type="pil"),
88
  gr.Slider(0, 1, value=0.5, label="Confidence", info="Show predictions that are above this confidence level")],
89
  outputs=gr.Label(),
90
- examples=[["pastel_yb.png", 0.5], ["piebald.png", 0.5], ["leopard_fire.png", 0.5]],
91
  title='Ball Python Morph Identifier',
92
  description="Upload or paste an image of your ball python to identify its morphs!"
93
  ).launch()
 
49
  checkpoint = torch.load(model_path, map_location=device)
50
 
51
  new_layers = nn.Sequential(
52
+ nn.LazyLinear(2048),
53
+ nn.BatchNorm1d(2048),
54
+ nn.ReLU(),
55
+ nn.Dropout(0.5),
56
+ nn.LazyLinear(num_labels)
57
+ )
58
 
59
  IMAGE_SIZE = 480
60
  transform = v2.Compose([
61
+ v2.ToImage(),
62
+ v2.Resize((IMAGE_SIZE, IMAGE_SIZE)),
63
+ v2.ToDtype(torch.float32),
64
+ v2.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
65
+ ])
66
 
67
  efficientnet = models.efficientnet_v2_l(weights='EfficientNet_V2_L_Weights.DEFAULT')
68
  efficientnet.classifier = new_layers
 
87
  inputs=[gr.Image(type="pil"),
88
  gr.Slider(0, 1, value=0.5, label="Confidence", info="Show predictions that are above this confidence level")],
89
  outputs=gr.Label(),
90
+ examples=[["enchi_albino_clown.png", 0.5], ["mojave_ghi.png", 0.5], ["hypo_banana_pastel_enchi.png", 0.5]],
91
  title='Ball Python Morph Identifier',
92
  description="Upload or paste an image of your ball python to identify its morphs!"
93
  ).launch()