fix img clf

Dockerfile.app

@@ -0,0 +1,2 @@
+FROM huggingface/autotrain-advanced:latest
+CMD uvicorn autotrain.app:app --host 0.0.0.0 --port 7860 --reload --workers 4

src/autotrain/cli/run_image_classification.py

@@ -280,7 +280,7 @@ def run(self):
                 valid_split=self.args.valid_split,
                 image_column=self.args.image_column,
                 target_column=self.args.target_column,
-                model_name=self.args.model,
+                model=self.args.model,
                 lr=self.args.lr,
                 epochs=self.args.epochs,
                 batch_size=self.args.batch_size,

src/autotrain/trainers/image_classification/__main__.py

@@ -175,7 +175,7 @@ def train(config):
 
 
 if __name__ == "__main__":
-    args = parse_args()
-    training_config = json.load(open(args.training_config))
-    config = ImageClassificationParams(**training_config)
-    train(config)
+    _args = parse_args()
+    training_config = json.load(open(_args.training_config))
+    _config = ImageClassificationParams(**training_config)
+    train(_config)