Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|