Update caption.py

caption.py

@@ -14,6 +14,9 @@ with gr.Blocks() as demo:
         [label]
     )
 
+if __name__ == '__main__':
+    demo.launch(share=True)
+
 model = AutoModelForCausalLM.from_pretrained("Chesscorner/git-chess-v3")
 processor = AutoProcessor.from_pretrained("Chesscorner/git-chess-v3")
 
@@ -42,6 +45,3 @@ def predict_step(image):
     # Decode predictions
     preds = processor.batch_decode(output_ids, skip_special_tokens=True)
     return preds[0].strip()
-
-if __name__ == '__main__':
-    demo.launch(share=True)