tiny update

app.py

@@ -24,6 +24,7 @@ class GradioApp:
         # Lazy loading of models
         if isinstance(self.models[model_name], str):
             self.models[model_name] = torch.load(self.models[model_name], map_location='cpu')
+            self.models[model_name].eval()
 
         img = torch.unsqueeze(self.models[model_name].val_transform(Image.open(img_file)), 0)
         with torch.inference_mode():