Ahsamkk commited on
Commit
f2ecfd4
·
verified ·
1 Parent(s): 43353fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,11 +1,15 @@
1
  from flask import Flask, request, jsonify
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
 
4
 
5
  app = Flask(__name__)
6
 
 
 
 
7
  # Load model once at startup
8
- model_name = "ahsamkk/urdu-sentiment-model"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)
10
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
11
 
 
1
  from flask import Flask, request, jsonify
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import torch
4
+ import os
5
 
6
  app = Flask(__name__)
7
 
8
+ # ✅ Fix cache permission issue
9
+ os.environ["HF_HOME"] = "/app/hf_cache"
10
+
11
  # Load model once at startup
12
+ model_name = "ahsamkk/urdu-sentiment-model" # your model name
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
15