Update app.py

app.py

@@ -7,20 +7,20 @@ import tempfile
 # DO NOT create a global client for paid apps
 # Each user's client will be created using their OAuth token
 
-def generate_video_with_auth(image, prompt, profile: gr.OAuthProfile | None, progress=gr.Progress()):
+def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None, progress=gr.Progress()):
     """
     Generate a video from an image using the Ovi model with authentication check.
     
     Args:
         image: Input image (PIL Image or file path)
         prompt: Text prompt describing the desired motion/animation
-        profile: OAuth profile for authentication
+        token: OAuth token for authentication
         progress: Gradio progress tracker
     
     Returns:
         Path to the generated video file
     """
-    if profile is None:
+    if token is None or not getattr(token, "token", None):
         raise gr.Error("Please sign in with Hugging Face to use this paid app")
     
     if image is None:
@@ -49,7 +49,7 @@ def generate_video_with_auth(image, prompt, profile: gr.OAuthProfile | None, pro
         # Do NOT use bill_to parameter - this makes the USER pay, not HuggingFace
         client = InferenceClient(
             provider="fal-ai",
-            api_key=profile.oauth_info.access_token,
+            api_key=token.token,  # Use token.token
         )
         
         # Generate video using the inference client