update app.py

app.py

@@ -12,7 +12,7 @@ import gradio as gr
 
 num_classes = len(TabletPeriodDataset.PERIOD_INDICES)
 
-class_weights = torch.load("data/class_weights_period.pt")
+class_weights = torch.load("class_weights_period.pt")
 
 checkpoint_path = 'epoch=22-step=213621.ckpt'
 vae_model = VAE.load_from_checkpoint(checkpoint_path,image_channels=1,z_dim=12, lr =0.0001, use_classification_loss=True, num_classes=num_classes,