Spaces:

Tousifahamed
/

smol-lm2-demo

Running

Tousifahamed commited on Jan 22

Commit

4d798b4

verified ·

1 Parent(s): e1354e9

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def load_model(checkpoint_path):
     return model
 # Load the model
-model = load_model("checkpoint_5050_quantized.pt")
 # Function to generate text
 def generate_text(prompt, max_length=50, temperature=1.0, top_k=50):

     return model
 # Load the model
+model = load_model("checkpoint_quantized.pt")
 # Function to generate text
 def generate_text(prompt, max_length=50, temperature=1.0, top_k=50):