commit from ssq

app.py

@@ -6,12 +6,8 @@ from skimage import img_as_ubyte
 
 from Allweather.util import load_img, save_img
 from basicsr.models.archs.histoformer_arch import Histoformer
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-# True
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
-# Tesla T4
 
-model_restoration = Histoformer.from_pretrained("sunsean/Histoformer-real").to("cpu")
+model_restoration = Histoformer.from_pretrained("sunsean/Histoformer-real")
 
 model_restoration.eval()