Update app.py

app.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 
 # Check if CUDA is available
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-model_path = "X:clip_logs/f1/f2/sb/2024_05_19-02_16_40-model_ViT-B-32-lr_5e-06-b_16-j_4-p_amp/checkpoints/epoch_20.pt"
+model_path = "ViT-B-32.pt"
 model_name = "ViT-B-32"
 
 # Load model and tokenizer