Update app.py

app.py

@@ -5,9 +5,6 @@ from clarifai_grpc.grpc.api.status import status_code_pb2
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-#torch.set_default_device("cpu")
-model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True).to(device)
-tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
 
 # GPT-4 credentials
 PAT_GPT4 = "3ca5bd8b0f2244eb8d0e4b2838fc3cf1"
@@ -38,6 +35,13 @@ MODEL_ID_NEWSGUARDIAN = "your_model_id"
 MODEL_VERSION_ID_NEWSGUARDIAN = "your_model_version_id"
 
 #
+import streamlit as st
+from clarifai_grpc.channel.clarifai_channel import ClarifaiChannel
+from clarifai_grpc.grpc.api import resources_pb2, service_pb2, service_pb2_grpc
+from clarifai_grpc.grpc.api.status import status_code_pb2
+
+
+
 
 # Set up gRPC channel for NewsGuardian model
 channel_tts = ClarifaiChannel.get_grpc_channel()
@@ -225,4 +229,9 @@ st.markdown("""
         <img src="https://user-images.githubusercontent.com/34125851/226601355-ffe0b597-9840-4e10-bbef-43d6c74b5a9e.png" width="2%" alt="" /></a>      
   </div>
   <hr>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)
+
+
+
+
+