Spaces:
Running
Running
raj-tomar001
commited on
Commit
•
a7856d0
1
Parent(s):
789d8ce
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def load_model(model_name):
|
|
16 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
17 |
return pipeline('text-classification', model=model, tokenizer=tokenizer, truncation=True, max_length=512, top_k=4)
|
18 |
|
19 |
-
classifier = load_model("./
|
20 |
|
21 |
# Load sample essays
|
22 |
with open(SAMPLE_JSON_PATH, 'r') as f:
|
|
|
16 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
17 |
return pipeline('text-classification', model=model, tokenizer=tokenizer, truncation=True, max_length=512, top_k=4)
|
18 |
|
19 |
+
classifier = load_model("./fine_tuned_roberta-base")
|
20 |
|
21 |
# Load sample essays
|
22 |
with open(SAMPLE_JSON_PATH, 'r') as f:
|