Hugging Face's logo

Spaces:
ethzanalytics
/
dialog-China
Runtime error

App Files Community
jonathanlehner commited on
Commit
630d42a
•
1 Parent(s): 8c7c98a

medium gave me strange results

Browse files
Files changed (1) hide show
  1. ai_single_response.py +1 -1
ai_single_response.py CHANGED
@@ -51,7 +51,7 @@ def query_gpt_model(
51
  [dict]: [returns a dict with A) just model response as str B) total conversation]
52
  """
53
  ai = aitextgen(
54
- model="microsoft/DialoGPT-medium",
55
  #model_folder=folder_path,
56
  to_gpu=False,
57
  )
 
51
  [dict]: [returns a dict with A) just model response as str B) total conversation]
52
  """
53
  ai = aitextgen(
54
+ model="microsoft/DialoGPT-large",
55
  #model_folder=folder_path,
56
  to_gpu=False,
57
  )