Update app.py

app.py

@@ -1,10 +1,10 @@
 from fastapi import FastAPI
 import uvicorn
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModel, AutoTokenizer
 
 model_name = "Tap-M/Luna-AI-Llama2-Uncensored"
 
-model = AutoModelForCausalLM.from_pretrained(
+model = AutoModel.from_pretrained(
     model_name,  # Example model
     offload_folder="./offload",  # Temporary directory
     trust_remote_code=True        # Required for some models