main_service.py

import gradio as gr
from utils_model import ModelFactory

def retrieve_embeddings(input_text_query, model_type):
    model_factory_obj = ModelFactory()
    model = model_factory_obj.create_model(model_type)

    query_embeddings = model.retrieve_embeddings(input_text_query)

    return query_embeddings

def build():
    models_list = ['all-MiniLM-L6-v2', 'sentence_similarity_spanish']

    app = gr.Interface(fn=retrieve_embeddings, 
                       inputs=["text", gr.Dropdown(models_list, label='Model type')], 
                       outputs="dataframe")

    return app

def run(app):
    app.launch(server_name='0.0.0.0')

app = build()
run(app)