Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import pandas as pd
 from io import StringIO
 import json
-import torch
+#import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM #AutoModelForTokenClassification
 from sentence_transformers import SentenceTransformer, util 
 #import lmdeploy
@@ -66,14 +66,8 @@ HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
 HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
 HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
 
-model_id = "meta-llama/Llama-3.2-1B-Instruct"
-pipe = pipeline(
-    "text-generation",
-    model=model_id,
-    torch_dtype=torch.bfloat16,
-    device_map="auto",
-)
-
+#model_id = "meta-llama/Llama-3.2-1B-Instruct"
+#pipe = pipeline("text-generation", model=model_id, torch_dtype=torch.bfloat16, device_map="auto",)
 
 
 col1, col2, col3 = st.columns([1,1,2.5])