Add files via upload

Author: Yash-xoxo

Transformers/Clear_chat.py

@@ -0,0 +1,26 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+# Load a pretrained model and tokenizer
+model_name = "microsoft/DialoGPT-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Chatbot interaction loop
+print("AI Chatbot ready! Type 'quit' to exit.")
+
+while True:
+    user_input = input("You: ")
+    if user_input.lower() == 'quit':
+        print("Goodbye!")
+        break
+
+    # Encode the user input
+    input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
+
+    # Generate a response
+    response_ids = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    print(f"Bot: {response}")
+

Transformers/Conversational_Model.py

@@ -0,0 +1,33 @@
+import openai
+
+openai.api_key = 'your-api-key'
+
+def generate_response(messages):
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        max_tokens=150
+    )
+    return response['choices'][0]['message']['content']
+
+# Chatbot interaction loop
+print("AI Chatbot ready! Type 'quit' to exit.")
+chat_history = []
+
+while True:
+    user_input = input("You: ")
+    if user_input.lower() == 'quit':
+        print("Goodbye!")
+        break
+
+    # Add user input to chat history
+    chat_history.append({"role": "user", "content": user_input})
+
+    # Generate a response
+    response = generate_response(chat_history)
+
+    # Add bot response to chat history
+    chat_history.append({"role": "assistant", "content": response})
+
+    print(f"Bot: {response}")
+

Transformers/Limit_chat.py

@@ -0,0 +1,35 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+# Load a pretrained model and tokenizer
+model_name = "microsoft/DialoGPT-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Chatbot interaction loop
+print("AI Chatbot ready! Type 'quit' to exit.")
+chat_history = []
+
+while True:
+    user_input = input("You: ")
+    if user_input.lower() == 'quit':
+        print("Goodbye!")
+        break
+
+    # Add user input to chat history, limit size to last 3 exchanges
+    chat_history.append(user_input)
+    if len(chat_history) > 6:  # Each exchange includes both user and bot messages
+        chat_history = chat_history[-6:]
+
+    # Join the chat history and encode it
+    input_ids = tokenizer.encode(" ".join(chat_history) + tokenizer.eos_token, return_tensors='pt')
+
+    # Generate a response
+    response_ids = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    # Add bot response to chat history
+    chat_history.append(response)
+
+    print(f"Bot: {response}")
+

Transformers/bot.py

@@ -0,0 +1,33 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+# Load a pretrained model and tokenizer
+model_name = "microsoft/DialoGPT-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Chatbot interaction loop
+print("AI Chatbot ready! Type 'quit' to exit.")
+chat_history = []
+
+while True:
+    user_input = input("You: ")
+    if user_input.lower() == 'quit':
+        print("Goodbye!")
+        break
+
+    # Add user input to chat history
+    chat_history.append(user_input)
+
+    # Join the chat history and encode it as input for the model
+    input_ids = tokenizer.encode(" ".join(chat_history) + tokenizer.eos_token, return_tensors='pt')
+
+    # Generate a response
+    response_ids = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    response = tokenizer.decode(response_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    # Add bot response to chat history
+    chat_history.append(response)
+
+    print(f"Bot: {response}")
+