Akjava commited on
Commit
6d9676e
Β·
verified Β·
1 Parent(s): 2229639

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ if not is_hugging_face:
57
 
58
  print("initialized")
59
 
60
- @spaces.GPU(duration=60)
61
  def generate_text(messages):
62
  if is_hugging_face:#need everytime initialize for ZeroGPU
63
  model = AutoModelForCausalLM.from_pretrained(
@@ -80,7 +80,7 @@ def generate_text(messages):
80
 
81
 
82
 
83
-
84
  def call_generate_text(message, history):
85
  # history.append({"role": "user", "content": message})
86
  #print(message)
 
57
 
58
  print("initialized")
59
 
60
+
61
  def generate_text(messages):
62
  if is_hugging_face:#need everytime initialize for ZeroGPU
63
  model = AutoModelForCausalLM.from_pretrained(
 
80
 
81
 
82
 
83
+ @spaces.GPU(duration=60)
84
  def call_generate_text(message, history):
85
  # history.append({"role": "user", "content": message})
86
  #print(message)