Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import gradio as gr
-from datasets import load_dataset, Dataset  # Added Dataset import
+from datasets import load_dataset, Dataset  
 from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,