alperugurcan commited on
Commit
fd3a527
·
verified ·
1 Parent(s): 6eba496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
- from transformers import PreTrainedTokenizerFast, PreTrainedModel
3
  import torch
4
  import re
5
 
6
  # Model ve tokenizer'ı yükleyin
7
- model_name = "alperugurcan/nlp-disaster" # Model dosyalarının bulunduğu dizini buraya yazın
8
- tokenizer = PreTrainedTokenizerFast.from_pretrained(model_name) # Tokenizer'ı yükleyin
9
- model = PreTrainedModel.from_pretrained(model_name) # Modeli yükleyin (modelin hangi sınıf olduğuna bağlı olarak değişebilir)
10
 
11
  def clean_text(text):
12
  # URL'leri ve özel karakterleri kaldır
 
1
  import gradio as gr
2
+ from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
3
  import torch
4
  import re
5
 
6
  # Model ve tokenizer'ı yükleyin
7
+ model_name = "your_model_directory" # Model dosyalarının bulunduğu dizini buraya yazın
8
+ tokenizer = DistilBertTokenizer.from_pretrained(model_name) # Doğru tokenizer sınıfını kullanın
9
+ model = DistilBertForSequenceClassification.from_pretrained(model_name) # Doğru model sınıfını kullanın
10
 
11
  def clean_text(text):
12
  # URL'leri ve özel karakterleri kaldır