Upload app.py

app.py

@@ -48,29 +48,29 @@ def setup(
         
         return None, None, description
 
-    try:
-        model = AutoModelForCausalLM.from_pretrained(
-            model_id, 
-            torch_dtype=torch.bfloat16, 
-            device_map="auto"
-        )
 
-        tokenizer = AutoTokenizer.from_pretrained(
-            model_id
-        )
+    model = AutoModelForCausalLM.from_pretrained(
+        model_id, 
+        torch_dtype=torch.bfloat16, 
+        device_map="auto"
+    )
 
-        tokenizer.use_default_system_prompt = False
-        
-        # Update the description
-        description += "\n<p>Model and tokenizer set up successfully.</p>"
-        
-        return model, tokenizer, description
+    tokenizer = AutoTokenizer.from_pretrained(
+        model_id
+    )
+
+    tokenizer.use_default_system_prompt = False
+    
+    # Update the description
+    description += "\n<p>Model and tokenizer set up successfully.</p>"
     
-    except Exception as e:
-        # If an error occurs during setup, append the error message to the description
-        description += f"\n<p>Error occurred during model setup: {str(e)}</p>"
+    return model, tokenizer, description
+    
+    # except Exception as e:
+    #     # If an error occurs during setup, append the error message to the description
+    #     description += f"\n<p>Error occurred during model setup: {str(e)}</p>"
         
-        return None, None, description
+    #     return None, None, description
 
 DESCRIPTION = """\
 # Pearl-7B-0211-ties, an xtraordinary 7B model