Commit
•
7fe75d1
1
Parent(s):
411d4a4
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ model = PeftModel.from_pretrained(model, adapter_model)
|
|
17 |
tokenizer = AutoTokenizer.from_pretrained(adapter_model)
|
18 |
|
19 |
def greet(text):
|
|
|
20 |
|
21 |
model.to('cpu')
|
22 |
inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
|
|
|
17 |
tokenizer = AutoTokenizer.from_pretrained(adapter_model)
|
18 |
|
19 |
def greet(text):
|
20 |
+
|
21 |
|
22 |
model.to('cpu')
|
23 |
inputs = tokenizer.encode(text, return_tensors="pt").to("cpu")
|