EngReem85 commited on
Commit
12964eb
·
verified ·
1 Parent(s): 863b886

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -4,7 +4,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
4
  model_name = "MahmoudIbrahim/Meta-LLama3-Instruct-Arabic"
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
- model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu")
 
 
 
 
 
8
 
9
 
10
  generator = pipeline(
 
4
  model_name = "MahmoudIbrahim/Meta-LLama3-Instruct-Arabic"
5
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForCausalLM.from_pretrained(
8
+ model_name,
9
+ device_map="cpu",
10
+ trust_remote_code=True
11
+ )
12
+
13
 
14
 
15
  generator = pipeline(