Update main.py

main.py

@@ -7,27 +7,9 @@ from sentence_transformers import SentenceTransformer, util
 
 app = FastAPI()
 current_directory = os.getcwd()
-os.environ['TRANSFORMERS_CACHE'] = current_directory
 
-# Function to load or download the SentenceTransformer model
-def load_model():
-    try:
-        model_path = os.path.join(current_directory, 'sentence-transformers/all-MiniLM-L6-v2')
-        
-        # Check if the model is already downloaded
-        if not os.path.exists(model_path):
-            # If not, download the model
-            model = SentenceTransformer('all-MiniLM-L6-v2', cache_folder=current_directory)
-        else:
-            # If the model is already downloaded, load it
-            model = SentenceTransformer(model_path, cache_folder=current_directory)
-        
-        return model
-    except Exception as e:
-        raise HTTPException(status_code=500, detail="An error occurred during model loading")
-
-# Load or download the SentenceTransformer model
-model = load_model()
+# Load the SentenceTransformer model
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
 class StringInput(BaseModel):
     sentence1: str
@@ -93,6 +75,7 @@ async def calculate_cosine_similarity(data: StringInput):
         return {"cosine_similarity": cos_sim}
     except Exception as e:
         raise HTTPException(status_code=500, detail="An error occurred during calculation")
+    
 
 if __name__ == "__main__":
-    uvicorn.run(app, port=8000)
+    uvicorn.run(app,port=8000)