yazied49 commited on
Commit
ebb39a6
·
verified ·
1 Parent(s): c37afa0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,21 +1,20 @@
1
  import os
 
 
2
  from fastapi import FastAPI
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
- # ✅ حل مشكلة الصلاحيات للكاش
6
- os.environ['TRANSFORMERS_CACHE'] = './cache'
7
-
8
  app = FastAPI()
9
 
10
  model_name = "microsoft/BioGPT-Large"
11
 
12
- # تحميل النموذج والمفردات
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  model = AutoModelForCausalLM.from_pretrained(model_name)
15
 
16
  @app.get("/")
17
  def read_root():
18
- return {"message": "BioGPT is running!"}
19
 
20
  @app.post("/generate")
21
  def generate_text(prompt: str):
 
1
  import os
2
+ os.environ['TRANSFORMERS_CACHE'] = './cache' # تغيير مكان الكاش
3
+
4
  from fastapi import FastAPI
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
 
 
 
7
  app = FastAPI()
8
 
9
  model_name = "microsoft/BioGPT-Large"
10
 
11
+ # تحميل النموذج والمفردات
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  model = AutoModelForCausalLM.from_pretrained(model_name)
14
 
15
  @app.get("/")
16
  def read_root():
17
+ return {"message": "BioGPT is ready!"}
18
 
19
  @app.post("/generate")
20
  def generate_text(prompt: str):