Spaces:

MahatirTusher
/

PneumoInsight

Sleeping

MahatirTusher commited on Feb 23, 2025

Commit

055fc26

verified ·

1 Parent(s): b06cc6f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ processor = ViTImageProcessor.from_pretrained(model_name_or_path)
 # Load dataset (adjust dataset_path accordingly)
 dataset_path = "pawlo2013/chest_xray"
-download_config = DownloadConfig(timeout=100, max_retries=10)
 train_dataset = load_dataset(dataset_path, split="train", download_config=download_config)
 class_names = train_dataset.features["label"].names

 # Load dataset (adjust dataset_path accordingly)
 dataset_path = "pawlo2013/chest_xray"
+download_config = DownloadConfig(max_retries=10)
 train_dataset = load_dataset(dataset_path, split="train", download_config=download_config)
 class_names = train_dataset.features["label"].names