From af20983d31c580e16c87bc231da05d9153c7492d Mon Sep 17 00:00:00 2001
From: zakaria47fs <55286921+zakaria47fs@users.noreply.github.com>
Date: Fri, 24 Jan 2020 13:15:38 +0100
Subject: [PATCH] Update Model.py

in the line 30 Avatar_type section takes directly 'head' as avatar type, so that I commented it and uncommented lines 24 to 28
---
 models/Model_AVATAR/Model.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/models/Model_AVATAR/Model.py b/models/Model_AVATAR/Model.py
index 3aef73d..ad49999 100644
--- a/models/Model_AVATAR/Model.py
+++ b/models/Model_AVATAR/Model.py
@@ -21,13 +21,13 @@ class AVATARModel(ModelBase):
     #override
     def onInitializeOptions(self, is_first_run, ask_override):
         if is_first_run:
-            #avatar_type = io.input_int("Avatar type ( 0:source, 1:head, 2:full_face ?:help skip:1) : ", 1, [0,1,2],
-            #                           help_message="Training target for the model. Source is direct untouched images. Full_face or head are centered nose unaligned faces.")
-            #avatar_type = {0:'source',
-            #               1:'head',
-            #               2:'full_face'}[avatar_type]
+            avatar_type = io.input_int("Avatar type ( 0:source, 1:head, 2:full_face ?:help skip:1) : ", 1, [0,1,2],
+                                       help_message="Training target for the model. Source is direct untouched images. Full_face or head are centered nose unaligned faces.")
+            avatar_type = {0:'source',
+                           1:'head',
+                           2:'full_face'}[avatar_type]
 
-            self.options['avatar_type'] = 'head'
+            #self.options['avatar_type'] = 'head'
         else:
             self.options['avatar_type'] = self.options.get('avatar_type', 'head')
 
@@ -487,4 +487,4 @@ class AVATARModel(ModelBase):
 
         return func
 
-Model = AVATARModel
\ No newline at end of file
+Model = AVATARModel