Upload app.py with huggingface_hub

app.py

@@ -13,8 +13,8 @@ def get_pipeline():
     if _PIPE is not None or _ERR is not None:
         return _PIPE, _ERR
     try:
-        tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-        model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
+        model = AutoModelForCausalLM.from_pretrained(MODEL_ID, trust_remote_code=True, torch_dtype="auto")
         _PIPE = TextGenerationPipeline(model=model, tokenizer=tokenizer, device=-1)
     except Exception as e:
         _ERR = str(e)