Functionality rework

models.py

@@ -23,6 +23,7 @@ class GigaChat:
 
     @classmethod
     def get_giga(cls, auth_file='auth_token.json'):
+        print('got giga')
         return cls(auth_file)
 
     def gen_giga_token(self, auth_file):
@@ -73,6 +74,7 @@ class GigaChat:
 
 
 def get_tinyllama():
+    print('got llama')
     tinyllama = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", torch_dtype=torch.float16, device_map="auto")
     return tinyllama
 

utils.py

@@ -17,6 +17,7 @@ def model_response(
         model_name,
         model_params
         ):
+    print(f'{model_name} response')
     chat_history.append([content])
     res = gen_lib[model_name](initialized_models[model_name], chat_history, model_params)
     chat_history[-1].append(res)