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(
|