Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ token = os.environ['TOKEN']
|
|
14 |
# Load the pretrained model and tokenizer
|
15 |
MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
|
16 |
|
17 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME
|
18 |
-
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME
|
19 |
|
20 |
# Predefined examples
|
21 |
examples = [
|
|
|
14 |
# Load the pretrained model and tokenizer
|
15 |
MODEL_NAME = "BounharAbdelaziz/Al-Atlas-LLM-0.5B" #"atlasia/Al-Atlas-LLM"
|
16 |
|
17 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)#, token=token)
|
18 |
+
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
|
19 |
|
20 |
# Predefined examples
|
21 |
examples = [
|