version changes

app.py

@@ -18,10 +18,10 @@ lora_model_id = "guimCC/segformer-v0-gta-cityscapes"
 original_model = SegformerForSemanticSegmentation.from_pretrained(original_model_id).to(device)
 lora_model = SegformerForSemanticSegmentation.from_pretrained(lora_model_id).to(device)
 
-# Load the dataset and select 10 random images
+# Load the dataset and select the first 10 images
 dataset = load_dataset("Chris1/cityscapes", split="validation")
-#sampled_dataset = random.sample(list(dataset), 10)  # Select 10 random examples
-sampled_dataset = dataset[:10]  # Select the first 10 examples
+sampled_dataset = dataset.select(range(10))  # Select the first 10 examples
+
 
 # Define your custom image processor
 jitter = ColorJitter(brightness=0.25, contrast=0.25, saturation=0.25, hue=0.1)

requirements.txt

@@ -1,5 +1,5 @@
 gradio
-transformers
+transformers==0.4.3
 torch
 pillow
 datasets