Abdulvahap commited on
Commit
c32326e
1 Parent(s): 3103fe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -8,8 +8,10 @@ RETRY_ATTEMPTS = 10
8
  TIMEOUT = 20 # Increase the timeout to 20 seconds
9
 
10
  # Load model directly
11
- from transformers import AutoModel
12
- model = AutoModel.from_pretrained("dbmdz/bert-base-turkish-128k-uncased")
 
 
13
 
14
  def predict(input_text):
15
  """Generate a response based solely on the current input text."""
 
8
  TIMEOUT = 20 # Increase the timeout to 20 seconds
9
 
10
  # Load model directly
11
+ from transformers import AutoTokenizer, AutoModel
12
+
13
+ tokenizer = AutoTokenizer.from_pretrained("emrecan/bert-base-turkish-cased-mean-nli-stsb-tr")
14
+ model = AutoModel.from_pretrained("emrecan/bert-base-turkish-cased-mean-nli-stsb-tr")
15
 
16
  def predict(input_text):
17
  """Generate a response based solely on the current input text."""