Update app.py

app.py

@@ -77,7 +77,7 @@ class S3DirectStream:
     def load_model_from_stream(self, model_prefix):
         try:
             logger.info(f"Loading model {model_prefix}...")
-            if self.file_exists_in_s3(f"{model_prefix}/config.json"): #Only check for config.json
+            if self.file_exists_in_s3(f"{model_prefix}/config.json"):
                 logger.info(f"Model {model_prefix} found in S3. Loading...")
                 return self.load_model_from_existing_s3(model_prefix)
 
@@ -97,9 +97,6 @@ class S3DirectStream:
         logger.info(f"Config loaded for {model_prefix}.")
 
         model_files = self._get_model_files(model_prefix) #This line is not used anymore
-        if not model_files:
-            logger.error(f"No model files found for {model_prefix} in S3")
-            raise EnvironmentError(f"No model files found for {model_prefix} in S3")
 
         state_dict = {}
         for model_file in model_files: #This loop is not used anymore
@@ -171,6 +168,7 @@ class S3DirectStream:
             logger.error(f"An unexpected error occurred: {e}")
             raise HTTPException(status_code=500, detail=f"An unexpected error occurred during model download") from e
 
+
     def download_and_upload_to_s3_url(self, url, s3_key):
         logger.info(f"Downloading from {url}...")
         with requests.get(url, stream=True) as response: