Spaces:

ssirikon
/

Gradio2-SubjectGen

Runtime error

ssirikon commited on Aug 2

Commit

d3c5c8a

•

1 Parent(s): 7bae295

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 # Replace with your model name
-#MODEL_NAME = "ssirikon/Gemma7b-bnb-Unsloth"
-MODEL_NAME = "unsloth/gemma-7b-bnb-4bit"
 # Load the model and tokenizer
 model = AutoModelForCausalLM.from_pretrained(

 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 # Replace with your model name
+MODEL_NAME = "ssirikon/Gemma7b-bnb-Unsloth"
+#MODEL_NAME = "unsloth/gemma-7b-bnb-4bit"
 # Load the model and tokenizer
 model = AutoModelForCausalLM.from_pretrained(