Update handler.py

handler.py

@@ -7,7 +7,7 @@ class CustomModelHandler:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.load_model()
 
-   def load_model(self):
+    def load_model(self):
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, trust_remote_code=True)
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name_or_path, 
@@ -27,4 +27,4 @@ handler = CustomModelHandler("microsoft/Phi-3-vision-128k-instruct")
 
 # Example prediction function
 def predict(input_text):
-    return handler.predict(input_text)
+    return handler.predict(input_text)