Add application file

app.py

@@ -2,7 +2,10 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-model_id = "t-tech/T-pro-it-1.0"
+# Установка сида для воспроизводимости
+torch.manual_seed(42)
+
+model_id = "t-tech/T-lite-it-1.0"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
@@ -12,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
 
 def generate_response(prompt):
     messages = [
-        {"role": "system", "content": "Ты T-pro, виртуальный ассистент в Т-Технологии. Твоя задача - быть полезным диалоговым ассистентом."},
+        {"role": "system", "content": "Ты T-lite, виртуальный ассистент в Т-Технологии. Твоя задача - быть полезным диалоговым ассистентом."},
         {"role": "user", "content": prompt}
     ]
     
@@ -40,7 +43,7 @@ interface = gr.Interface(
     fn=generate_response,
     inputs="text",
     outputs="text",
-    title="T-pro API"
+    title="T-lite API"
 )
 
 interface.launch()