app v1

README.md

@@ -1,13 +1,18 @@
 ---
-title: Diego Llm Chat DialoGPT
-emoji: 💻
-colorFrom: yellow
-colorTo: indigo
+title: Diego's LLM Chat
+emoji: 🤖
 sdk: gradio
-sdk_version: 4.29.0
+sdk_version: 4.24.0
+license: cc-by-nc-sa-4.0
+short_description: DialoGPT LLM model chat
+colorFrom: red
+colorTo: indigo
 app_file: app.py
-pinned: false
-license: unlicense
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+### Result
+* LLM Chat
+* Using LLM Model DialoGPT
+
+<img src='result.png' />
+

app.py

@@ -0,0 +1,37 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+
+# Load DialoGPT model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+
+def text_to_chat(initial_text, question):
+    # Concatenate the initial text and the question
+    text = initial_text + " " + question
+
+    # Encode the new user input, add the eos_token and return a tensor in Pytorch
+    new_user_input_ids = tokenizer.encode(text + tokenizer.eos_token, return_tensors='pt')
+
+    # Generate a response to the user input
+    bot_input_ids = model.generate(new_user_input_ids, max_length=1024, pad_token_id=tokenizer.eos_token_id, max_new_tokens=500)
+
+    # Decode the response and return it
+    chat_output = tokenizer.decode(bot_input_ids[:, new_user_input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    return chat_output
+
+def chat_with_bot(initial_text, question):
+    result = text_to_chat(initial_text, question)
+    return result
+
+ui = gr.Interface(fn=chat_with_bot, 
+                     inputs=["text", "text"], 
+                     outputs="text",
+                     title="Diego's LLM Chat with DialoGPT",
+                     examples=
+                            [["","What is the capital of France?"],
+                            ["A man walked into a bar.", "What happened next?"],
+                            ["Acording to NoOne, The purpose of life is about the journey to learn and improve or just 42", "Acording to NoOne, what is the purpose of life?"]],
+                     description="DialoGPT is a large-scale pretrained dialogue response generation model. It is based on the GPT-2 architecture with minor modifications to the training objective. The model is trained on multiple tasks, including conversational response generation, and can generate coherent and contextually relevant responses to user inputs.",
+                    )
+ui.launch()

install-deps.sh

@@ -0,0 +1,3 @@
+#!/bin/bash 
+
+pip install -r requirements.txt

requirements.txt

@@ -0,0 +1,18 @@
+numpy
+transformers
+sentence-transformers
+seaborn
+torch
+torchvision
+matplotlib
+pandas
+scikit-learn
+nltk
+gensim
+tensorflow
+keras
+opencv-python
+fastapi
+uvicorn
+gradio
+flask

run.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+python app.py