Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,8 @@ import streamlit as st
|
|
| 5 |
from profanity_check import predict
|
| 6 |
|
| 7 |
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# Chat Template
|
| 10 |
|
|
@@ -26,7 +28,8 @@ llm = HuggingFacePipeline.from_model_id(
|
|
| 26 |
model_id='TinyLlama/TinyLlama-1.1B-Chat-v1.0',
|
| 27 |
task='text-generation',
|
| 28 |
pipeline_kwargs=dict(
|
| 29 |
-
max_new_tokens = 512
|
|
|
|
| 30 |
)
|
| 31 |
)
|
| 32 |
|
|
|
|
| 5 |
from profanity_check import predict
|
| 6 |
|
| 7 |
|
| 8 |
+
st.set_page_config(page_title="Chef Medi", layout="centered")
|
| 9 |
+
st.title("Chef Medi")
|
| 10 |
|
| 11 |
# Chat Template
|
| 12 |
|
|
|
|
| 28 |
model_id='TinyLlama/TinyLlama-1.1B-Chat-v1.0',
|
| 29 |
task='text-generation',
|
| 30 |
pipeline_kwargs=dict(
|
| 31 |
+
max_new_tokens = 512,
|
| 32 |
+
temperature = 0.1
|
| 33 |
)
|
| 34 |
)
|
| 35 |
|