Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,6 +13,8 @@ tokenizer = AutoTokenizer.from_pretrained("Cogwisechat/falcon-7b-finance")
|
|
13 |
|
14 |
|
15 |
def generate_text(input_text):
|
|
|
|
|
16 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
17 |
attention_mask = torch.ones(input_ids.shape)
|
18 |
|
@@ -26,7 +28,6 @@ def generate_text(input_text):
|
|
26 |
eos_token_id=tokenizer.eos_token_id,
|
27 |
)
|
28 |
|
29 |
-
global cleaned_output_text
|
30 |
|
31 |
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
32 |
print(output_text)
|
@@ -46,7 +47,7 @@ with block:
|
|
46 |
message = gr.Textbox(placeholder='Enter Your Question Here')
|
47 |
state = gr.State()
|
48 |
submit = gr.Button("SEND")
|
49 |
-
submit.click(generate_text, inputs=[message, state], outputs=[
|
50 |
|
51 |
block.launch(debug = True)
|
52 |
|
|
|
13 |
|
14 |
|
15 |
def generate_text(input_text):
|
16 |
+
global output_text
|
17 |
+
|
18 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
19 |
attention_mask = torch.ones(input_ids.shape)
|
20 |
|
|
|
28 |
eos_token_id=tokenizer.eos_token_id,
|
29 |
)
|
30 |
|
|
|
31 |
|
32 |
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
33 |
print(output_text)
|
|
|
47 |
message = gr.Textbox(placeholder='Enter Your Question Here')
|
48 |
state = gr.State()
|
49 |
submit = gr.Button("SEND")
|
50 |
+
submit.click(generate_text, inputs=[message, state], outputs=[output_text, state])
|
51 |
|
52 |
block.launch(debug = True)
|
53 |
|