Spaces:
Sleeping
Sleeping
iamsubrata
commited on
Commit
•
de70828
1
Parent(s):
160dee1
fix button bug
Browse files
LLAMA2.py
CHANGED
@@ -16,7 +16,7 @@ text="""4Bit QLoRA Fine-Tuned LLM on English Quotes Dataset."""
|
|
16 |
|
17 |
st.caption(text)
|
18 |
|
19 |
-
|
20 |
label = "Prompt...",
|
21 |
value = "Tell me a joke."
|
22 |
)
|
@@ -29,10 +29,10 @@ LLM = pipeline(
|
|
29 |
)
|
30 |
|
31 |
if text_input:
|
32 |
-
|
33 |
label = "Generate",
|
34 |
key = "generate"
|
35 |
)
|
36 |
|
37 |
if prompt:
|
38 |
-
st.write(LLM(str(
|
|
|
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)))
|