Update app.py

app.py

@@ -4,7 +4,7 @@ import logging
 import numpy as np
 from sentence_transformers import SentenceTransformer
 import torch
-from torch.cuda.amp import autocast
+from torch.amp import autocast  # Corrected import for autocast
 from spaces import GPU
 import json  # Import json for direct JSON output in UI
 
@@ -51,7 +51,7 @@ def generate_embedding(text, focus):
         initialize_model()
 
     try:
-        with autocast("cuda"):
+        with torch.amp.autocast('cuda'): # Corrected autocast usage
             embedding_vector = model.encode([text])[0].tolist() # Get embedding as list
         # Convert embedding to JSON string for direct display in UI
         embedding_json_str = json.dumps(embedding_vector)
@@ -98,7 +98,7 @@ def process_files(files, focus):
             try:
                 with open(file.name, 'r') as f:
                     text = f.read()
-                with autocast("cuda"):
+                with torch.amp.autocast('cuda'): # Corrected autocast usage
                     embedding = model.encode([text])[0].tolist()
                 all_embeddings.append(embedding)
                 file_statuses.append(f"File '{file.name}' processed successfully.")