Andrey Vorozhko commited on
Commit
e87e0fe
·
1 Parent(s): 8571554

Import necessary functions

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import random
3
  import torch
4
  import gradio as gr
5
  from transformers import AutoModelForCausalLM, AutoTokenizer
6
- # from util_funcs import getLengthParam, calcAnswerLengthByProbability, cropContext
7
 
8
  def chat_function(Message, History): # model, tokenizer
9
 
@@ -124,4 +124,4 @@ iface = gr.Interface(fn=chat_function,
124
  )
125
 
126
  if __name__ == "__main__":
127
- iface.launch(debug=True)
 
3
  import torch
4
  import gradio as gr
5
  from transformers import AutoModelForCausalLM, AutoTokenizer
6
+ from util_funcs import getLengthParam, calcAnswerLengthByProbability, cropContext
7
 
8
  def chat_function(Message, History): # model, tokenizer
9
 
 
124
  )
125
 
126
  if __name__ == "__main__":
127
+ iface.launch(debug=True, share=True)