KvrParaskevi commited on
Commit
9326be5
1 Parent(s): ce4746c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  def model_wrapper(text):
5
  try:
6
  # Load the model with custom code execution allowed
7
- model = gr.load("models/cerebras/btlm-3b-8k-chat", trust_remote_code=True)
8
  # Use the model to predict
9
  return model(text)
10
  except Exception as e:
 
4
  def model_wrapper(text):
5
  try:
6
  # Load the model with custom code execution allowed
7
+ model = gr.load("models/cerebras/btlm-3b-8k-chat")
8
  # Use the model to predict
9
  return model(text)
10
  except Exception as e: