Skip to content

Commit 562df76

Browse files
author
CORRUPTOR2037
committed
fixed bottom of the chatbox going under input textfield
1 parent 5a36fca commit 562df76

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

webui/app.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
import sys
22
import os
33
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
4-
from model import ExLlama, ExLlamaConfig
54
from flask import Flask, render_template, request, jsonify
65
from flask import Response, stream_with_context
76
from threading import Timer, Lock
87
import webbrowser
98
import json
109
import model_init
11-
from session import prepare_sessions, get_initial_session, Session, load_session, new_session, _sessions_dir
10+
from session import prepare_sessions, get_generator, get_initial_session, Session, load_session, new_session, _sessions_dir
1211
import argparse
1312
from tokenizer import ExLlamaTokenizer
13+
from model import ExLlama, ExLlamaConfig
1414
from waitress import serve
1515

1616
app = Flask(__name__)
@@ -139,12 +139,21 @@ def api_append_block():
139139
data = request.get_json()
140140
session.api_append_block(data)
141141
return json.dumps({"result": "ok"}) + "\n"
142+
143+
@app.route('/inference', methods=['POST'])
144+
def inferContextP():
145+
prompt = request.form.get('prompt')
146+
tokens = int(request.form.get('max_tokens', 200))
147+
148+
outputs = get_generator().generate_simple(prompt, max_new_tokens = tokens)
149+
print(outputs)
150+
return outputs
142151

143152
# Load the model
144153

145154
parser = argparse.ArgumentParser(description="Simple web-based chatbot for ExLlama")
146155
parser.add_argument("-host", "--host", type = str, help = "IP:PORT eg, 0.0.0.0:7862", default = "localhost:5000")
147-
parser.add_argument("-sd", "--sessions_dir", type = str, help = "Location for storing user sessions, default: ~/exllama_sessions/", default = "~/exllama_sessions/")
156+
parser.add_argument("-sd", "--sessions-dir", type = str, help = "Location for storing user sessions, default: ~/exllama_sessions/", default = "~/exllama_sessions/")
148157

149158
model_init.add_args(parser)
150159
args = parser.parse_args()

webui/static/style.css

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,8 @@ svg:hover {
181181
padding-top: 10px;
182182
padding-bottom: 10px;
183183
box-sizing: border-box;
184+
top: 0px;
185+
position: fixed;
184186
}
185187

186188
#chat-column {

0 commit comments

Comments
 (0)