Arrcttacsrks commited on
Commit
7ade650
·
verified ·
1 Parent(s): ffec595

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
  app = Flask(__name__)
7
 
8
  # Khởi tạo mô hình
9
- model_id = "meta-llama/Llama-3.2-3B-Instruct"
10
  pipe = pipeline("text-generation", model=model_id, torch_dtype=torch.bfloat16)
11
 
12
  @app.route('/')
 
6
  app = Flask(__name__)
7
 
8
  # Khởi tạo mô hình
9
+ model_id = "Arrcttacsrks/Llama3.2-3B-Instruct"
10
  pipe = pipeline("text-generation", model=model_id, torch_dtype=torch.bfloat16)
11
 
12
  @app.route('/')