Internal changes

PiperOrigin-RevId: 615088561
techthiyanes · Mar 12, 2024 · dc93911 · dc93911
1 parent 98f3a63
commit dc93911
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/langfun/core/modalities/image.py b/langfun/core/modalities/image.py
@@ -34,7 +34,7 @@ def mime_type(self) -> str:
     return f'image/{self.image_format}'
 
   def _repr_html_(self) -> str:
-    if self.uri:
+    if self.uri and self.uri.lower().startswith(('http:', 'https:', 'ftp:')):
       return f'<img src="{self.uri}">'
     image_raw = base64.b64encode(self.to_bytes()).decode()
     return f'<img src="data:image/{self.image_format};base64,{image_raw}">'