Spaces:

AbhishekBhavnani
/

TweetClassification

Sleeping

AbhishekBhavnani commited on Apr 22

Commit

9e99fa9

verified ·

1 Parent(s): 581d7d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ model_ckpt = "AbhishekBhavnani/TweetClassification"
 model = AutoModelForSequenceClassification.from_pretrained(
     model_ckpt,
-    low_cpu_mem_usage=True  # To prevent large memory usage during loading
 )
 tokenizer = AutoTokenizer.from_pretrained(model_ckpt)

 model = AutoModelForSequenceClassification.from_pretrained(
     model_ckpt,
+    low_cpu_mem_usage=False
 )
 tokenizer = AutoTokenizer.from_pretrained(model_ckpt)