Spaces:

AIFS
/

Prometh-MOEM-24B

Runtime error

Wtzwho commited on Feb 16

Commit

9c1f9ef

•

1 Parent(s): 1c63b9f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ text_generation_pipeline = pipeline(
     model_kwargs={"torch_dtype": torch.float16, "load_in_4bit": True},
     use_auth_token=hf_token
 )
-)
 def generate_text(user_input):
     messages = [{"role": "user", "content": user_input}]

     model_kwargs={"torch_dtype": torch.float16, "load_in_4bit": True},
     use_auth_token=hf_token
 )
 def generate_text(user_input):
     messages = [{"role": "user", "content": user_input}]