Update app.py

app.py

@@ -12,12 +12,11 @@ import soundfile as sf
 import numpy as np
 from fastapi import FastAPI, HTTPException, Request, UploadFile, File
 from fastapi.responses import StreamingResponse
-from pydantic import BaseModel, ValidationError, field_validator
+from pydantic import BaseModel, field_validator
 from transformers import (
     AutoConfig,
     AutoModelForCausalLM,
     AutoTokenizer,
-    pipeline,
     GenerationConfig,
     StoppingCriteriaList
 )
@@ -110,7 +109,7 @@ class S3ModelLoader:
     def download_model_from_huggingface(self, model_name):
         try:
             logging.info(f"Downloading model {model_name} from Hugging Face...")
-            model_dir = hf_hub_download(model_name, token=HUGGINGFACE_HUB_TOKEN)
+            model_dir = hf_hub_download(model_name, token=HUGGINGFACE_HUB_TOKEN, filename=model_name.split('/')[-1])
             self.s3_client.upload_file(model_dir, self.bucket_name, f"lilmeaty_garca/{model_name}")
             logging.info(f"Model {model_name} saved to S3 successfully.")
         except Exception as e:
@@ -222,8 +221,6 @@ async def generate(request: Request, body: GenerateRequest):
 
     except HTTPException as e:
         raise e
-    except ValidationError as e:
-        raise HTTPException(status_code=422, detail=e.errors())
     except Exception as e:
         logging.exception(f"An unexpected error occurred: {e}")
         raise HTTPException(status_code=500, detail="An unexpected error occurred.")