VanguardAI commited on
Commit
4a19484
·
verified ·
1 Parent(s): 2398182

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -53,6 +53,7 @@ alpaca_prompt = """Below is an instruction that describes a task, paired with an
53
 
54
  @spaces.GPU(duration=300)
55
  def chunk_it(inventory_list, user_input_text):
 
56
  inputs = tokenizer(
57
  [
58
  alpaca_prompt.format(
 
53
 
54
  @spaces.GPU(duration=300)
55
  def chunk_it(inventory_list, user_input_text):
56
+ model.to('cuda')
57
  inputs = tokenizer(
58
  [
59
  alpaca_prompt.format(