Update app.py

app.py

@@ -1,10 +1,4 @@
-# -*- coding: utf-8 -*-
-"""gemma_3n_colab.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1U5pbaYG8qD7HFANwU7PI1jbLGOPvArnP
+# -*- coding: utf-8 
 
 # 🏥 Gemma 3N SOAP Note Generator
 ## Interactive Medical Documentation Assistant
@@ -12,11 +6,6 @@ Original file is located at
 This notebook provides a complete interface for generating SOAP notes from medical text using the Gemma 3N model.
 """
 
-# Install required packages
-!pip install -q transformers torch torchvision torchaudio timm accelerate
-!pip install -q ipywidgets gradio
-!pip install -q --upgrade huggingface_hub
-!pip install GPUtil
 
 # Enable widgets
 from IPython.display import display, HTML
@@ -34,18 +23,35 @@ from datetime import datetime
 from huggingface_hub import login
 import getpass
 
+# Authenticate with HuggingFace
+# Replace the authentication section (lines around the getpass part) with this:
+
+# Import libraries and authenticate
+import torch
+from transformers import AutoProcessor, AutoModelForImageTextToText
+import gradio as gr
+import ipywidgets as widgets
+from IPython.display import display, clear_output
+import io
+import base64
+from datetime import datetime
+from huggingface_hub import login
+import os
+
 # Authenticate with HuggingFace
 print("🔐 HuggingFace Authentication Required")
-print("Please enter your HuggingFace token (it will be hidden):")
-hf_token = getpass.getpass("HF Token: ")
 
+# Try to get token from environment variable first (for production/HF Spaces)
+hf_token = os.environ.get('HF_TOKEN') or os.environ.get('HUGGINGFACE_TOKEN')
+
+if hf_token:
+    print("✅ Found HF token in environment variables")
 try:
     login(token=hf_token)
     print("✅ Successfully authenticated with HuggingFace!")
 except Exception as e:
     print(f"❌ Authentication failed: {e}")
     print("Please check your token and try again.")
-
 # Check GPU availability
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"🖥️  Using device: {device}")
@@ -703,4 +709,3 @@ show_system_info()
 *🤖 Powered by Google's Gemma 3N Model | 🔒 All processing performed locally*
 """
 
-!gradio deploy