Update models.py

models.py

@@ -1,9 +1,21 @@
+python
+
+Executar
+
+Copiar
 import os
+import shutil
 from transformers import LlamaTokenizer, LlamaForCausalLM
 import torch
 
+# Definir o diretório de cache para o modelo
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/model_cache'
 
+# Limpar o diretório de cache, se existir
+cache_dir = '/tmp/model_cache'
+if os.path.exists(cache_dir):
+    shutil.rmtree(cache_dir)
+
 def carregar_modelo():
     tokenizer = LlamaTokenizer.from_pretrained("lzw1008/Emollama-7b")
     model = LlamaForCausalLM.from_pretrained("lzw1008/Emollama-7b", device_map="auto")