updated req.txt and app.py

README.md

@@ -6,7 +6,7 @@ colorTo: green
 sdk: gradio
 sdk_version: 4.19.2
 python_version: 3.8
-app_file: app_old.py
+app_file: app.py
 pinned: false
 ---
 

app.py

@@ -2,6 +2,7 @@ from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel, PeftConfig
 import gradio as gr
+import torch
 
 
 base_model = "unsloth/Llama-3.2-3B-Instruct"  # Replace with the correct base model