Spaces:

Vishwas1
/

HFCompareModel

Sleeping

Vishwas1 commited on Sep 23, 2024

Commit

30579a3

verified ·

1 Parent(s): 03b46bf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import os
 # Load the original pre-trained model
 def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return tokenizer, model
 # Models to compare

 # Load the original pre-trained model
 def load_model(model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
     return tokenizer, model
 # Models to compare