Spaces:

67Ayush87
/

DataScience

Running

67Ayush87 commited on 18 days ago

Commit

004eb6e

verified ·

1 Parent(s): 1a317f9

Update pages/gen_ai.py

Files changed (1) hide show

pages/gen_ai.py CHANGED Viewed

@@ -23,7 +23,7 @@ genai_skeleton = HuggingFaceEndpoint(
     repo_id='google/gemma-2-9b-it',
     provider='nebius',
     temperature=0.7,
-    max_new_tokens=150,
     task='conversational'
 )
@@ -32,7 +32,7 @@ genai_chat = ChatHuggingFace(
     repo_id='google/gemma-2-9b-it',
     provider='nebius',
     temperature=0.7,
-    max_new_tokens=150,
     task='conversational'
 )

     repo_id='google/gemma-2-9b-it',
     provider='nebius',
     temperature=0.7,
+    max_new_tokens=50,
     task='conversational'
 )
     repo_id='google/gemma-2-9b-it',
     provider='nebius',
     temperature=0.7,
+    max_new_tokens=50,
     task='conversational'
 )