Spaces:

prithivMLmods
/

Pocket-Callisto

Running on Zero

prithivMLmods commited on 9 days ago

Commit

2c8490a

verified ·

1 Parent(s): 9810ea7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def generate_reply(model, tokenizer, prompt, max_length, do_sample):
     )
     return tokenizer.decode(output_tokens[0], skip_special_tokens=True)
-@sapces.GPU
 def main():
     args, _ = get_args()
     model, tokenizer = load_model(args.model)

     )
     return tokenizer.decode(output_tokens[0], skip_special_tokens=True)
+@spaces.GPU
 def main():
     args, _ = get_args()
     model, tokenizer = load_model(args.model)