Add application file

app.py

@@ -22,8 +22,8 @@ app.add_middleware(
 
 # Load GIT-base model and processor
 print("🚀 Loading microsoft/git-base-coco model...")
-processor = GitProcessor.from_pretrained("microsoft/git-base-coco")
-model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco", torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
+processor = GitProcessor.from_pretrained("microsoft/git-base-coco/tree/main")
+model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco/tree/main", torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)
 print("✅ Model loaded!")
 
 @app.get("/")