Update main.py

main.py

@@ -90,8 +90,9 @@ class QuranRecitationComparer:
         try:
             if token:
                 print(f"Loading model {model_name} with token...")
-                self.processor = AutoProcessor.from_pretrained(model_name, token=token)
-                self.model = AutoModelForCTC.from_pretrained(model_name, token=token)
+                # Use 'use_auth_token' instead of the deprecated 'token' parameter
+                self.processor = AutoProcessor.from_pretrained(model_name, use_auth_token=token)
+                self.model = AutoModelForCTC.from_pretrained(model_name, use_auth_token=token)
             else:
                 print(f"Loading model {model_name} without token...")
                 self.processor = AutoProcessor.from_pretrained(model_name)
@@ -99,7 +100,7 @@ class QuranRecitationComparer:
                 
             self.model = self.model.to(self.device)
             self.model.eval()
-            # Set the configuration to always return hidden states
+            # Ensure that hidden states are returned by default
             self.model.config.output_hidden_states = True
             print("Model loaded successfully!")
         except Exception as e:
@@ -142,7 +143,7 @@ class QuranRecitationComparer:
             ).input_values.to(self.device)
             
             with torch.no_grad():
-                # Call the model without passing output_hidden_states explicitly.
+                # Call the model without explicitly passing output_hidden_states
                 outputs = self.model(inputs)
             
             hidden_states = outputs.hidden_states[-1]
@@ -278,10 +279,7 @@ async def startup_event():
 @app.get("/")
 async def root():
     """Root endpoint to check if the API is running."""
-    if comparer:
-        status = "active"
-    else:
-        status = "model not loaded"
+    status = "active" if comparer else "model not loaded"
     return {"message": "Quran Recitation Comparer API is running", "status": status}
 
 @app.post("/compare", response_model=ComparisonResult)