Jessica Walkenhorst commited on
Commit
dde2901
1 Parent(s): 586ec44

Update description

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ description = """Hans Christian Andersen's tale of the ugly duckling tells us ab
16
 
17
  examples = ['duckling.jpg', 'cygnet.jpg']
18
 
19
- article = 'This model was build using a resnet-18 architecture with weights pretrained on the ImageNet data set and fine-tuned using about 80 images of ducklings and cygnets each.\nNote that it is binary classifier and can therefore only output cygnet or duckling, "other" is not an option.'
20
 
21
  app = gr.Interface(fn=classify_image,
22
  inputs=gr.components.Image(),
 
16
 
17
  examples = ['duckling.jpg', 'cygnet.jpg']
18
 
19
+ article = '**Technical Details**: The classification model was build using a resnet-18 architecture. Training was done using a transfer learning approach. I took the publicly available weights that were pre-trained on the ImageNet data set and fine-tuned them using about 80 images of ducklings and cygnets each.\nNote that it is binary classifier and will therefore only output "cygnet" or "duckling", "other" is not an option.'
20
 
21
  app = gr.Interface(fn=classify_image,
22
  inputs=gr.components.Image(),