iamsubrata commited on
Commit
de70828
1 Parent(s): 160dee1

fix button bug

Browse files
Files changed (1) hide show
  1. LLAMA2.py +3 -3
LLAMA2.py CHANGED
@@ -16,7 +16,7 @@ text="""4Bit QLoRA Fine-Tuned LLM on English Quotes Dataset."""
16
 
17
  st.caption(text)
18
 
19
- text_input = st.text_area(
20
  label = "Prompt...",
21
  value = "Tell me a joke."
22
  )
@@ -29,10 +29,10 @@ LLM = pipeline(
29
  )
30
 
31
  if text_input:
32
- prompt = st.button(
33
  label = "Generate",
34
  key = "generate"
35
  )
36
 
37
  if prompt:
38
- st.write(LLM(str(prompt)))
 
16
 
17
  st.caption(text)
18
 
19
+ user_input = st.text_area(
20
  label = "Prompt...",
21
  value = "Tell me a joke."
22
  )
 
29
  )
30
 
31
  if text_input:
32
+ button = st.button(
33
  label = "Generate",
34
  key = "generate"
35
  )
36
 
37
  if prompt:
38
+ st.write(LLM(str(user_input)))