IlyaGusev commited on
Commit
ba177eb
1 Parent(s): b19f953

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -49,6 +49,9 @@ def load_model(
49
  return model
50
 
51
 
 
 
 
52
  def user(message, history):
53
  new_history = history + [[message, None]]
54
  return "", new_history
@@ -61,6 +64,7 @@ def bot(
61
  top_k,
62
  temp
63
  ):
 
64
  tokens = get_system_tokens(model)[:]
65
 
66
  for user_message, bot_message in history[:-1]:
 
49
  return model
50
 
51
 
52
+ MODEL = load_model()
53
+
54
+
55
  def user(message, history):
56
  new_history = history + [[message, None]]
57
  return "", new_history
 
64
  top_k,
65
  temp
66
  ):
67
+ model = MODEL
68
  tokens = get_system_tokens(model)[:]
69
 
70
  for user_message, bot_message in history[:-1]: