Update app.py

app.py

@@ -1,61 +0,0 @@
-import streamlit as st
-from langchain_community.document_loaders import YoutubeLoader
-from transformers import AutoTokenizer, AutoModelForCausalLM
-import torch
-
-st.set_page_config(page_title="DeepSeek YouTube Summarizer", layout="centered")
-st.title("📺 YouTube Video Summarizer with DeepSeek")
-
-# Input YouTube URL
-url = st.text_input("Enter YouTube Video URL:")
-
-# Load DeepSeek model and tokenizer
-@st.cache_resource
-def load_model():
-    model_id = "deepseek-ai/deepseek-llm-7b-instruct"
-    tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
-    model = AutoModelForCausalLM.from_pretrained(
-        model_id, device_map="auto", torch_dtype=torch.float16, trust_remote_code=True
-    )
-    return tokenizer, model
-
-tokenizer, model = load_model()
-
-def summarize_with_deepseek(text):
-    prompt = f"""<|system|>\nYou are a helpful assistant.\n<|user|>\nSummarize the following text:\n{text}\n<|assistant|>"""
-    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
-    outputs = model.generate(
-        **inputs,
-        max_new_tokens=300,
-        do_sample=False,
-        temperature=0.7,
-        top_k=50,
-        top_p=0.95,
-    )
-    summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return summary.split("<|assistant|>")[-1].strip()
-
-if st.button("Extract and Summarize"):
-    if url:
-        try:
-            loader = YoutubeLoader.from_youtube_url(url)
-            data = loader.load()
-            transcript = data[0].page_content if data else "No transcript found."
-
-            st.subheader("📖 Extracted Transcript")
-            st.text_area("Transcript:", transcript, height=300)
-
-            # Truncate for prompt length safety
-            if len(transcript) > 1500:
-                transcript = transcript[:1500]
-
-            with st.spinner("Summarizing using DeepSeek..."):
-                summary = summarize_with_deepseek(transcript)
-
-            st.subheader("🧠 Summary")
-            st.success(summary)
-
-        except Exception as e:
-            st.error(f"Error: {str(e)}")
-    else:
-        st.warning("Please enter a valid YouTube URL.")