Spaces:

Pclanglais
/

Editorialization

Running

Pclanglais commited on Jul 2

Commit

80a2f6a

•

1 Parent(s): 11b325d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import pandas as pd
 # Define the device
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_checkpoint = "PleIAs/Estienne"
 token_classifier = pipeline(
     "token-classification", model=editorial_model, aggregation_strategy="simple", device=device
 )

 # Define the device
 device = "cuda" if torch.cuda.is_available() else "cpu"
+editorial_model = "PleIAs/Estienne"
 token_classifier = pipeline(
     "token-classification", model=editorial_model, aggregation_strategy="simple", device=device
 )