Rename app.txt to app.py

app.txt → app.py

@@ -1,8 +1,8 @@
 import torch
 from transformers import AutoModelForTokenClassification,AutoTokenizer,pipeline
 
-model = AutoModelForTokenClassification.from_pretrained('HooshvareLab/bert-fa-base-uncased')
-tokenizer = AutoTokenizer.from_pretrained('HooshvareLab/bert-fa-base-uncased')
+model = AutoModelForTokenClassification.from_pretrained('SeyedAli/Persian-Text-NER-Bert-V1')
+tokenizer = AutoTokenizer.from_pretrained('SeyedAli/Persian-Text-NER-Bert-V1')
 
 def NER(text):
     pipline = pipeline(task="token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")