Spaces:
Sleeping
Sleeping
File size: 460 Bytes
c634a8f 32c15f7 11e25c5 b81a54b a742d90 b81a54b 32e6759 dacb0b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# 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)
|