Update app.py

app.py

@@ -1,3 +1,17 @@
+import sys
+import types
+
+torch_six = types.ModuleType('torch._six')
+torch_six.string_classes = (str,)  # In Python 3, string_classes is just (str,)
+
+# Create torch module if it doesn't exist in sys.modules
+if 'torch' not in sys.modules:
+    import torch
+    
+# Add the _six submodule to torch
+sys.modules['torch._six'] = torch_six
+torch._six = torch_six
+
 from pydoc import describe
 import gradio as gr
 import torch