Changed lightning version

app.py

@@ -209,7 +209,7 @@ if __name__ == "__main__":
 
     images_dir = glob(os.path.join(os.getcwd(), "samples") + os.sep + "*.png")
     examples = [[i, TrainingConfig.METRIC_THRESH] for i in np.random.choice(images_dir, size=8, replace=False)]
-    print(examples)
+    # print(examples)
 
     iface = gr.Interface(
         fn=partial(predict, model=model, preprocess_fn=preprocess, device=DEVICE, idx2labels=labels),

requirements.txt

@@ -1,6 +1,4 @@
---find-links https://download.pytorch.org/whl/torch_stable.html
-torch==2.0.0+cpu
-torchvision==0.15.0
-torchmetrics==1.0.0
-pydantic
-lightning
+torch==2.0.0
+torchvision
+lightning==2.0.1
+torchmetrics==1.0.0