Update app.py

app.py

@@ -2,7 +2,7 @@ import pathlib
 import gradio as gr
 import transformers
 from transformers import AutoTokenizer
-from transformers import ModelForCausalLM 
+from transformers import AutoModelForCausalLM 
 from transformers import GenerationConfig
 from typing import List, Dict, Union
 from typing import Any, TypeVar
@@ -10,7 +10,7 @@ from typing import Any, TypeVar
 Pathable = Union[str, pathlib.Path]
 
 def load_model(name: str) -> Any:
-    return ModelForCausalLM.from_pretrained(name)
+    return AutoModelForCausalLM.from_pretrained(name)
 
 def load_tokenizer(name: str) -> Any:
     return AutoTokenizer.from_pretrained(name)