Update app.py

app.py

@@ -5,11 +5,11 @@ import os
 
 os.system("git clone https://github.com/ai-forever/KandiSuperRes.git")
 os.system("cd KandiSuperRes")
-from KandiSuperRes import get_SR_pipeline
+from KandiSuperRes import sr_pipeline
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-sr_pipe2x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=2)
-sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=4)
+sr_pipe2x = sr_pipeline(device=device, fp16=True, flash=True, scale=2)
+sr_pipe4x = sr_pipeline(device=device, fp16=True, flash=True, scale=4)
 
 
 def inference(image, size):