Aytaj commited on
Commit
21ef1b2
1 Parent(s): 04ce092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -5,7 +5,6 @@ import torch
5
  # Load DialoGPT model and tokenizer
6
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
7
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
8
-
9
  # Function to generate a response
10
  def generate_response(chat_history_ids, new_user_input):
11
  new_user_input_ids = tokenizer.encode(new_user_input + tokenizer.eos_token, return_tensors='pt')
 
5
  # Load DialoGPT model and tokenizer
6
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
7
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
8
  # Function to generate a response
9
  def generate_response(chat_history_ids, new_user_input):
10
  new_user_input_ids = tokenizer.encode(new_user_input + tokenizer.eos_token, return_tensors='pt')