Spaces:

KGSAGAR
/

Hindi_Text_Normalization

Sleeping

KGSAGAR commited on Feb 10

Commit

bd08591

verified ·

1 Parent(s): e6f7aa3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ def respond(
     top_p,
     peft_model,
     tokenizer_name='your-tokenizer-name',
-    device='cpu'  # or 'cpu' based on your setup
 ):
     """
     Generates a response based on the user message and history using the provided PEFT model.

     top_p,
     peft_model,
     tokenizer_name='your-tokenizer-name',
+    device='auto'  # or 'cpu' based on your setup
 ):
     """
     Generates a response based on the user message and history using the provided PEFT model.