Update app.py

app.py

@@ -3,13 +3,11 @@ from PIL import Image
 import gradio as gr
 import os
 
-os.system("git clone https://github.com/ai-forever/KandiSuperRes.git .")
-os.system("ls")
-from KandiSuperRes import sr_pipeline
+from KandiSuperRes import get_SR_pipeline
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-sr_pipe2x = sr_pipeline(device=device, fp16=True, flash=True, scale=2)
-sr_pipe4x = sr_pipeline(device=device, fp16=True, flash=True, scale=4)
+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)
 
 
 def inference(image, size):