iamsubrata commited on
Commit
c4b0f88
1 Parent(s): de70828

fix button bug

Browse files
Files changed (1) hide show
  1. LLAMA2.py +6 -6
LLAMA2.py CHANGED
@@ -16,11 +16,6 @@ text="""4Bit QLoRA Fine-Tuned LLM on English Quotes Dataset."""
16
 
17
  st.caption(text)
18
 
19
- user_input = st.text_area(
20
- label = "Prompt...",
21
- value = "Tell me a joke."
22
- )
23
-
24
  model_id = "iamsubrata/Llama-2-7b-chat-hf-sharded-bf16-fine-tuned"
25
 
26
  LLM = pipeline(
@@ -28,7 +23,12 @@ LLM = pipeline(
28
  model = model_id
29
  )
30
 
31
- if text_input:
 
 
 
 
 
32
  button = st.button(
33
  label = "Generate",
34
  key = "generate"
 
16
 
17
  st.caption(text)
18
 
 
 
 
 
 
19
  model_id = "iamsubrata/Llama-2-7b-chat-hf-sharded-bf16-fine-tuned"
20
 
21
  LLM = pipeline(
 
23
  model = model_id
24
  )
25
 
26
+ user_input = st.text_area(
27
+ label = "Prompt...",
28
+ value = "Tell me a joke."
29
+ )
30
+
31
+ if user_input:
32
  button = st.button(
33
  label = "Generate",
34
  key = "generate"