# Week - 3 Assignment - Integrate Traditional Chatbot with AI Service Project (Transformers) Praveen Kumar Parimi

#importing the required libraries including transformers
import gradio as gr
from huggingface_hub import InferenceClient
from transformers import pipeline, AutoModel, AutoTokenizer

model_name = "bmas10/ForJerry"
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", model=model_name)
pipe(messages)