Andrey Vorozhko
commited on
Commit
·
e87e0fe
1
Parent(s):
8571554
Import necessary functions
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import random
|
|
3 |
import torch
|
4 |
import gradio as gr
|
5 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
6 |
-
|
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)
|