Spaces:

bhlewis
/

patent-bert-finetuning

Paused

bhlewis commited on Jul 29, 2024

Commit

7e8026c

verified ·

1 Parent(s): cbac82e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import gradio as gr
 import json
 from transformers import AutoModelForSequenceClassification, Trainer, TrainingArguments, AutoTokenizer
 from datasets import Dataset
 # Load tokenized data
 def load_data(tokenized_file):

 import json
 from transformers import AutoModelForSequenceClassification, Trainer, TrainingArguments, AutoTokenizer
 from datasets import Dataset
+import torch
 # Load tokenized data
 def load_data(tokenized_file):