Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import AutoModelForTokenClassification, AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForTokenClassification
 import torch
 from threading import Thread
 import uvicorn
@@ -11,8 +11,9 @@ import requests
 app = FastAPI()
 
 # Cargar el modelo y el tokenizador
-tokenizer = AutoTokenizer.from_pretrained("mdarhri00/named-entity-recognition")
-model = AutoModelForTokenClassification.from_pretrained("mdarhri00/named-entity-recognition")
+model_name = "mdarhri00/named-entity-recognition"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForTokenClassification.from_pretrained(model_name)
 
 class TextInput(BaseModel):
     text: str
@@ -53,4 +54,4 @@ def predict_gradio(text):
     return entities
 
 demo = gr.Interface(fn=predict_gradio, inputs="text", outputs="json")
-demo.launch(share=True)
+demo.launch(share=True)