Update app.py

app.py

@@ -1,9 +1,6 @@
-import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-from diffusers import StableDiffusionPipeline
-import torch
 import os
 import logging
+from huggingface_hub import login
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -13,12 +10,10 @@ logger = logging.getLogger(__name__)
 read_token = os.getenv('AccToken')
 if not read_token:
     raise ValueError("Hugging Face access token not found. Please set the AccToken environment variable.")
-from huggingface_hub import login
-login(read_token)
+logger.info(f"Hugging Face access token found: {read_token[:5]}...")  # Log the first 5 characters for verification
 
-# Set device to GPU if available
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-logger.info(f"Device set to use {device}")
+# Log in to Hugging Face using the token
+login(read_token)
 
 # Define a dictionary of conversational models
 conversational_models = {
@@ -130,8 +125,6 @@ def generate_speech(model_name, text):
     audio = pipeline(text, speaker_embeddings=speaker_embeddings)
     return audio["audio"]
 
-
-
 def visual_qa(image, question):
     result = visual_qa_pipeline(image, question)
     return result["answer"]