model.py

import gradio as gr
class Model:
    def __init__(self,
            name:str="Model"):
        self.name = name
        self.model = None
        self.tokenizer = None
        self.gen_config = None
        self.INPUT_FORMAT = None
    
    def generate(self, inputs:str) -> str:
        outputs = inputs
        return outputs
    
    def chat(self, message, chat_history):
        response = self.generate(message)
        chat_history.append((message, response))
        return "", gr.Chatbot(chat_history, label=self.name, bubble_full_width=False)