Spaces:

ikaruart
/

sarvam_malayalam

Runtime error

ikaruart commited on 12 days ago

Commit

e4707e5

verified ·

1 Parent(s): ec7840d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,7 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
 from flask import Flask, request
 from transformers import AutoTokenizer, AutoModelForCausalLM

 import os
+os.environ["HF_HOME"] = "/app/hf_cache"  # Or any writable folder in your container
+from transformers import AutoTokenizer
 from flask import Flask, request
 from transformers import AutoTokenizer, AutoModelForCausalLM