Update app.py

app.py

@@ -63,7 +63,10 @@ def codegemma(message: str,
         do_sample=True,
         temperature=temperature,
     )
-
+    # This will enforce greedy generation (do_sample=False) when the temperature is passed 0, avoiding the crash.             
+    if temperature == 0:
+        generate_kwargs['do_sample'] = False
+        
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()