Sandiago21 commited on
Commit
4b8ef5e
·
verified ·
1 Parent(s): 23291aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,6 +69,7 @@ model = AutoModelForCausalLM.from_pretrained(
69
  # device_map=config.DEVICE
70
  # )
71
 
 
72
  def generate(prompt):
73
  """
74
  Generate a text completion from a causal language model given a prompt.
@@ -975,7 +976,6 @@ class BasicAgent:
975
  if question != "aalskdalsdh" and filename == "":
976
  time.sleep(120)
977
 
978
-
979
  state = {
980
  "messages": question,
981
  }
 
69
  # device_map=config.DEVICE
70
  # )
71
 
72
+
73
  def generate(prompt):
74
  """
75
  Generate a text completion from a causal language model given a prompt.
 
976
  if question != "aalskdalsdh" and filename == "":
977
  time.sleep(120)
978
 
 
979
  state = {
980
  "messages": question,
981
  }