RashidNLP
/

German-Text-Classification

Text Classification

text-embeddings-inference

Model card Files Files and versions

RashidNLP commited on Jun 1, 2023

Commit

d96e871

·

1 Parent(s): 2384e58

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -7,7 +7,7 @@ tags:
 - Discourse Analysis
 - German text classification
 ---
-# Roberta for German Discourse Classification
 This is a xlm Roberta model finetuned on a German Discourse dataset of 60 discourses having a total over 10k sentences.
@@ -40,7 +40,7 @@ def get_label(sentence):
         bert_dict[keys[i]] = round(probs[i].item(), 3)
     return bert_dict
-MODEL_NAME = 'RashidNLP/Roberta-German-Discourse'
 MODEL_DIR = 'model'
 CHECKPOINT_DIR = 'checkpoints'
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

 - Discourse Analysis
 - German text classification
 ---
+# Roberta for German text Classification
 This is a xlm Roberta model finetuned on a German Discourse dataset of 60 discourses having a total over 10k sentences.
         bert_dict[keys[i]] = round(probs[i].item(), 3)
     return bert_dict
+MODEL_NAME = 'RashidNLP/German-Text-Classification'
 MODEL_DIR = 'model'
 CHECKPOINT_DIR = 'checkpoints'
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')