Spaces:

BICORP
/

server-5

Runtime error

BICORP commited on Feb 18

Commit

f3a90c3

verified ·

1 Parent(s): 3cf5c1f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 # Load model and tokenizer locally
-MODEL_NAME = "gpt2"  # Replace with your local model directory if different
 def load_model():
     try:

 import torch
 # Load model and tokenizer locally
+MODEL_NAME = "BICORP/Lake-1-12B-spe"  # Replace with your local model directory if different
 def load_model():
     try: