CogwiseAI commited on
Commit
58a98b6
·
1 Parent(s): dcb567b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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=[cleaned_output_text, state])
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