Update app.py

app.py

@@ -5,14 +5,14 @@ from libs.eurovoc.EurovocTh import EurovocTh
 
 api_key = environ.get("api_key")
 
-def app(input, filter_strategy, relevence_threshold, k, length_management_strategy, token_overlap, results_merge_stategy, drop_last):
+def app(input, filter_strategy, relevence_threshold, k, length_management_strategy, token_overlap, results_merge_stategy, drop_last, model_name):
 
     th = EurovocTh()
 
-    domain_classifier_pipeline = pipeline(model="gianma/classifierEUtopLevelLongerTrain", tokenizer="gianma/classifierEUtopLevelLongerTrain", use_auth_token=api_key)
+    domain_classifier_pipeline = pipeline(model=model_name, tokenizer=model_name, use_auth_token=api_key)
     length_limit = domain_classifier_pipeline.tokenizer.model_max_length
 
-    eu_th_classifier_pipeline = pipeline(model="gianma/classifierEUtopLevelLongerTrain", tokenizer='gianma/classifierEUtopLevelLongerTrain', use_auth_token=api_key)
+    eu_th_classifier_pipeline = pipeline(model=model_name, tokenizer=model_name, use_auth_token=api_key)
     length_limit = eu_th_classifier_pipeline.tokenizer.model_max_length
     
     kwargs = {'padding':True,'truncation':True,'max_length':length_limit}
@@ -207,7 +207,9 @@ with gr.Blocks() as interface:
       with gr.Column():
         outputs = gr.Label()
 
+    model_name = "gianma/classifierEUtopLevelLongerTrain"
+
     clear_button.click(fn=lambda: None, inputs=None, outputs=input_text)
-    classify_button.click(app, inputs=[input_text,filter_strategy, s_confidence, s_k, document_reading_strategy, chunk_overlap, combine_strategy, exclude_last], outputs=outputs)
+    classify_button.click(app, inputs=[input_text,filter_strategy, s_confidence, s_k, document_reading_strategy, chunk_overlap, combine_strategy, exclude_last, model_name], outputs=outputs)
 
 interface.launch()