Fixed header, added temperature slider

app.py

@@ -11,25 +11,25 @@ def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.json()
 
-def get_completion(prompt):
+def get_completion(prompt, temperature=0.6):
     output = query({
 	"inputs": f"{prompt}",
 	"parameters": {
-        "temperature":0.6,
+        "temperature":temperature,
     }
     })
     return output[0]["generated_text"]
 
 def main():
-    st.title('UiA AI Koordinator Llama 3 8b text completion')
+    st.title('UiA ai koordinator\n Llama 3 8b text completion')
 
-    # Text input
+    temperature = st.slider('Select the temperature:', 0.0, 1.0, 0.6)
     user_input = st.text_area("Enter your prompt:", height=300)
     
     with st.spinner('Generating completion...'):
         if st.button('Generate'):
             # Get the model's completion
-            completion = get_completion(user_input)
+            completion = get_completion(user_input, temperature)
             st.text_area("Model Completion:", value=completion, height=300, key="2")
 
 if __name__ == '__main__':