Spaces:

quelmap
/

apitest

Sleeping

quelmap commited on Aug 13, 2023

Commit

351535d

1 Parent(s): e808886

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ import gradio
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-large", device_map="auto", torch_dtype=torch.float32)
-tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-large")
 def my_inference_function(mail):
     mail=mail.replace("\n", "")

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-3b", device_map="auto", torch_dtype=torch.float32)
+tokenizer = AutoTokenizer.from_pretrained("cyberagent/open-calm-3b")
 def my_inference_function(mail):
     mail=mail.replace("\n", "")