Spaces:

DmitriySv
/

Ticket_Classifier

Sleeping

DmitriySv commited on Oct 5, 2024

Commit

aaa5966

verified ·

1 Parent(s): 7e8fde4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import BertTokenizer, BertForSequenceClassification, BertModel, BertConfig
 import torch
 class MultiTaskBertModel(torch.nn.Module):
     def __init__(self, bert_model, num_labels_task1, num_labels_task2):

 import gradio as gr
 from transformers import BertTokenizer, BertForSequenceClassification, BertModel, BertConfig
 import torch
+import os
 class MultiTaskBertModel(torch.nn.Module):
     def __init__(self, bert_model, num_labels_task1, num_labels_task2):