Docker deploy

caption_server.py

@@ -30,7 +30,7 @@ git_coco_pipeline = pipeline("image-to-text", model=git_model_name)
 # Initialize the model for semantic similarity
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
-classification_server_url = 'http://localhost:8001/classify_text'
+classification_server_url = 'https://raviipandey-classification.hf.space/classify_text'
 
 def calculate_similarity(caption1, caption2):
     embeddings = model.encode([caption1, caption2], convert_to_tensor=True)