Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 from pptx import Presentation
@@ -9,10 +8,7 @@ import io, tempfile, os, re
 st.set_page_config(page_title="PPTX Smart Enhancer", layout="wide")
 
 # Load your hosted Hugging Face model
-from transformers import LlamaForCausalLM
-
-model = LlamaForCausalLM.from_pretrained("Faisalkhany/llam1b_finetune")
-  # ← change this
+MODEL_ID = "Faisalkhany/llam1b_finetune"  # ← change this
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model     = AutoModelForCausalLM.from_pretrained(MODEL_ID, device_map="auto", torch_dtype="auto")
 pipe      = pipeline("text-generation", model=model, tokenizer=tokenizer)
@@ -135,4 +131,4 @@ def main():
             mime="application/vnd.openxmlformats-officedocument.presentationml.presentation")
 
 if __name__=="__main__":
-    main()
+    main()