Hugging Face's logo

Spaces:
sagaxlearn
/
api-smollm135m
Sleeping

App Files Community
khurrameycon commited on
Commit
83f8a3e
·
verified ·
1 Parent(s): 1749217

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,7 +11,9 @@ class ModelInput(BaseModel):
11
  app = FastAPI()
12
 
13
  # Load your model and tokenizer
14
- model_path = "khurrameycon/SmolLM-135M-Instruct-qa_pairs_converted.json-25epochs"
 
 
15
  tokenizer = AutoTokenizer.from_pretrained(model_path)
16
  model = AutoModelForCausalLM.from_pretrained(model_path)
17
 
 
11
  app = FastAPI()
12
 
13
  # Load your model and tokenizer
14
+ model_path = "HuggingFaceTB/SmolLM2-135M-Instruct"
15
+ # "khurrameycon/SmolLM-135M-Instruct-qa_pairs_converted.json-25epochs"
16
+
17
  tokenizer = AutoTokenizer.from_pretrained(model_path)
18
  model = AutoModelForCausalLM.from_pretrained(model_path)
19