Update app.py

app.py

@@ -11,9 +11,8 @@ import requests
 app = FastAPI()
 
 # Cargar el modelo y el tokenizador
-model_name = "mdarhri00/named-entity-recognition"
-model = AutoModelForTokenClassification.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained("mdarhri00/named-entity-recognition")
+model = AutoModelForTokenClassification.from_pretrained("mdarhri00/named-entity-recognition")
 
 class TextInput(BaseModel):
     text: str
@@ -54,4 +53,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)