RayanAli commited on
Commit
c3b5a73
·
verified ·
1 Parent(s): cb30900

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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