Spaces:
Configuration error
Configuration error
ToonTownTommy
commited on
Delete app.py
Browse files
app.py
DELETED
@@ -1,50 +0,0 @@
|
|
1 |
-
import os
|
2 |
-
import gradio as gr
|
3 |
-
import ctransformers
|
4 |
-
|
5 |
-
configObj = ctransformers.Config(stop=["\n", 'User'], context_length=2048)
|
6 |
-
config = ctransformers.AutoConfig(config=configObj, model_type='llama')
|
7 |
-
config.config.stop = ["\n"]
|
8 |
-
|
9 |
-
# path_to_llm = os.path.abspath("llama-2-7b-chat.ggmlv3.q4_1.bin")
|
10 |
-
|
11 |
-
llm = ctransformers.AutoModelForCausalLM.from_pretrained("ToonTownTommy/llama2-7b-chat-hf-GGML", model_file="llama2-7b-chat-hf-GGML", config=config)
|
12 |
-
|
13 |
-
def complete(prompt, stop=["User", "Assistant"]):
|
14 |
-
tokens = llm.tokenize(prompt)
|
15 |
-
output = ''
|
16 |
-
for token in llm.generate(tokens):
|
17 |
-
result = llm.detokenize(token)
|
18 |
-
output += result
|
19 |
-
for word in stop:
|
20 |
-
if word in output:
|
21 |
-
print('\n')
|
22 |
-
return output
|
23 |
-
print(result, end='',flush=True)
|
24 |
-
|
25 |
-
print('\n')
|
26 |
-
return output
|
27 |
-
|
28 |
-
title = "llama2-7b-chat-ggml"
|
29 |
-
description = "This space is an attempt to run the GGUF 4 bit quantized version of 'llama2-7b-chat' on a CPU"
|
30 |
-
|
31 |
-
example_1 = "Write a 7 line poem on AI"
|
32 |
-
example_2 = "Tell me a joke"
|
33 |
-
|
34 |
-
examples = [example_1, example_2]
|
35 |
-
|
36 |
-
def generate_response(user_input):
|
37 |
-
prompt = f'User: {user_input}\nAssistant: '
|
38 |
-
response = complete(prompt)
|
39 |
-
return response
|
40 |
-
|
41 |
-
UI = gr.Interface(
|
42 |
-
fn=generate_response,
|
43 |
-
inputs=gr.Textbox(label="User Query", placeholder="Ask your queries here...."),
|
44 |
-
outputs=gr.Textbox(label="Assistant Response"),
|
45 |
-
title=title,
|
46 |
-
description=description,
|
47 |
-
examples=examples
|
48 |
-
)
|
49 |
-
|
50 |
-
UI.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|