Update app.py

app.py

@@ -352,7 +352,7 @@ def validate_and_convert_image(image, target_size=(512 , 512)):
     return image
 
 
-class Framer:
+class Drag:
 
     def __init__(self, device, args, height, width, model_length, dtype=torch.float16, use_sift=False):
         self.device = device
@@ -547,7 +547,7 @@ with gr.Blocks() as demo:
                 3. Interpolate the images (according the path) with a click on "Run" button. <br>""")
     
     # device, args, height, width, model_length
-    Framer = Framer("cuda:0", args, 320, 512, 14)
+    Framer = Drag("cuda", args, 320, 512, 14)
     first_frame_path = gr.State()
     last_frame_path = gr.State()
     tracking_points = gr.State([])