Update app.py

app.py

@@ -10,9 +10,9 @@ import os
 token = os.getenv("modelo")
 login(token = token)
 
-question = "Name the planets in the solar system? A: "
-question = "Quais são os planetas do sistema solar?" 
-question = "Qual é o maior planeta do sistema solar?" 
+# question = "Name the planets in the solar system? A: "
+# question = "Quais são os planetas do sistema solar?" 
+# question = "Qual é o maior planeta do sistema solar?" 
 
 before = datetime.datetime.now()
 
@@ -30,30 +30,31 @@ pipeline = transformers.pipeline(
     device_map="auto",
 )
 
+question = None
+
 with st.container():
     st.write('\n\n')
     st.title('LLM-LANAChat\n\n')
     question = st.text_input("Chat", placeholder="E ai?")
 
- 
-
-    messages = [
-     # {"role": "system", "content": "You are a chatbot"},
-    {"role": "user", "content": question}
-    ]
-    st.write('Gerando a saída...')
-    outputs = pipeline(
-      messages,
-      max_new_tokens=256,
-    )
-    response = outputs[0]["generated_text"][-1]
-    st.write('Saída gerada...')
-
-
-# =====================================================
-
-    print('\n\n')
-    st.write(response)
+    if question is not None:
+        messages = [
+         # {"role": "system", "content": "You are a chatbot"},
+        {"role": "user", "content": question}
+        ]
+        st.write('Gerando a saída...')
+        outputs = pipeline(
+          messages,
+          max_new_tokens=256,
+        )
+        response = outputs[0]["generated_text"][-1]
+        st.write('Saída gerada...')
+    
+    
+    # =====================================================
+    
+        print('\n\n')
+        st.write(response)
 
 
     after = datetime.datetime.now()