Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,11 @@
+import os
+
+# Rediriger les caches vers un dossier accessible en écriture
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/.cache/huggingface/transformers'
+os.environ['HF_HOME'] = '/tmp/.cache/huggingface'
+os.environ['XDG_CACHE_HOME'] = '/tmp/.cache'
+os.environ['STREAMLIT_HOME'] = '/tmp/.streamlit'
+
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
@@ -9,11 +17,7 @@ st.title("🧠 Chatbot - OpenChat 3.5")
 def load_model():
     model_name = "openchat/openchat-3.5-0106"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(
-        model_name,
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-        device_map="auto"
-    )
+    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
     return tokenizer, model
 
 tokenizer, model = load_model()