foreisight commited on
Commit
a5c73f5
·
verified ·
1 Parent(s): 079a4c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,6 +2,9 @@ import os
2
  from flask import Flask, request, jsonify
3
  from transformers import MarianMTModel, MarianTokenizer
4
 
 
 
 
5
  model_name = "Helsinki-NLP/opus-mt-en-de"
6
  tokenizer = MarianTokenizer.from_pretrained(model_name)
7
  model = MarianMTModel.from_pretrained(model_name)
 
2
  from flask import Flask, request, jsonify
3
  from transformers import MarianMTModel, MarianTokenizer
4
 
5
+ # ✅ Set a writable cache directory
6
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
7
+
8
  model_name = "Helsinki-NLP/opus-mt-en-de"
9
  tokenizer = MarianTokenizer.from_pretrained(model_name)
10
  model = MarianMTModel.from_pretrained(model_name)