Update app.py

app.py

@@ -5,7 +5,7 @@ import PIL
 import gradio as gr
 import os
 
-# Load model
+# Load model 2
 checkpoint_path = "model_checkpoint.pth"
 checkpoint_and_model = model_from_checkpoint(checkpoint_path)
 model = checkpoint_and_model["model"]