Update app.py

app.py

@@ -32,14 +32,24 @@ def init_client():
         return None, str(e)
 
 def classify_text(client, text_content):
-    """Classify text using Google Cloud Natural Language API"""
+    """Classify text using Google Cloud Natural Language API with V2 model"""
     try:
         document = language_v1.Document(
             content=text_content, 
             type_=language_v1.Document.Type.PLAIN_TEXT
         )
         
-        response = client.classify_text(request={'document': document})
+        # Configure V2 model options for better performance
+        classification_model_options = language_v1.ClassificationModelOptions(
+            v2_model=language_v1.ClassificationModelOptions.V2Model()
+        )
+        
+        response = client.classify_text(
+            request={
+                'document': document,
+                'classification_model_options': classification_model_options
+            }
+        )
         categories = response.categories
         
         results = []