Update app.py

app.py

@@ -9,7 +9,7 @@ import spaces
 from io import BytesIO
 
 
-class Classifier:
+class Litton7Classifier:
     LABELS = [
         "Panoramic",
         "Feature",
@@ -29,7 +29,6 @@ class Classifier:
         else:
             self.device = device
 
-        self.device = device
         self.model = torch.load(
             model_path, map_location=self.device, weights_only=False
         )
@@ -149,7 +148,7 @@ def get_layout():
                 '<div class="footer">© 2024 LCL 版權所有<br>開發者：何立智、楊哲睿</div>',
             )
             start_button.click(
-                fn=Classifier().predict,
+                fn=Litton7Classifier().predict,
                 inputs=image_input,
                 outputs=chart,
             )