update 4 app.py

app.py

@@ -1,17 +1,8 @@
-import gradio as gr
+# Use a pipeline as a high-level helper
 from transformers import pipeline
 
-# Загрузка модели
-model = pipeline("conversational", model="LLM360/K2-Chat")
-
-def chat(message, history):
-    history.append((message, ""))
-    response = model(history)
-    history[-1] = (message, response[0]['generated_text'])
-    return history, ""
-
-# Создание интерфейса
-iface = gr.ChatInterface(chat)
-
-# Запуск интерфейса
-iface.launch()
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="LLM360/K2-Chat")
+pipe(messages)