Update app.py

app.py

@@ -1,16 +1,14 @@
 import transformers
 import torch
-import gradio as gr
 
-# Model setup
 model_id = "yodayo-ai/nephra_v1.0"
 
-# Load the model pipeline with the recommended settings
 pipeline = transformers.pipeline(
     "text-generation",
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto",
+    offload_folder="offload",  # Add this line to specify the offload folder
 )
 
 # Define characters