ShAnSantosh commited on
Commit
099c56e
1 Parent(s): be97f1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,9 +1,9 @@
1
- from transformers import AutoModelForCausalLM, AutoTokenizer
2
  import torch
3
-
4
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
5
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
6
-
7
  def predict(input, history=[]):
8
  """
9
  # tokenize the new input sentence
 
1
+ #from transformers import AutoModelForCausalLM, AutoTokenizer
2
  import torch
3
+ """
4
  tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
5
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
6
+ """
7
  def predict(input, history=[]):
8
  """
9
  # tokenize the new input sentence