Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 import spaces
 from transformers import pipeline
+import torch
 
+# Initialize model with explicit dtype
 model = pipeline(
     "text-generation",
     model="unsloth/DeepSeek-R1-Distill-Llama-8B",
     device_map="auto",
-    torch_dtype=torch.float16,  # Explicit dtype
-    model_kwargs={"load_in_8bit": True},  # Reduces VRAM usage
+    torch_dtype=torch.float16,  # Now recognizes 'torch'
+    model_kwargs={"load_in_8bit": True}
 )
 
 @spaces.GPU(duration=300)  # Increased to 5 minutes