fully-app first commit

README.md

@@ -11,3 +11,6 @@ license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+LIVE DEPLOYMENT AT: https://huggingface.co/spaces/mrdbourke/foodvision_big

train.py

@@ -33,7 +33,7 @@ if __name__ == "__main__":
     # print(model)
 
 
-    loss_fn = torch.nn.CrossEntropyLoss()
+    loss_fn = torch.nn.CrossEntropyLoss(label_smoothing=0.1)
     accuracy_fn = MulticlassAccuracy(num_classes = NUM_CLASSES).to(device)
     optimizer = torch.optim.Adam(model.parameters(), lr = LEARNIGN_RATE)