Update app.py

app.py

@@ -24,7 +24,7 @@ def codegemma(message: str, history: list, temperature: float, max_new_tokens: i
     Returns:
         str: The generated response.
     """
-    input_ids = tokenizer(message, return_tensors="pt")
+    input_ids = tokenizer(message, return_tensors="pt").to("cuda:0")
     outputs = model.generate(
         **input_ids,
         temperature=temperature,
@@ -35,13 +35,13 @@ def codegemma(message: str, history: list, temperature: float, max_new_tokens: i
 
     
 placeholder = """
-<img src="https://huggingface.co/spaces/ysharma/CodeGemma/resolve/main/gemma_lockup_vertical_full-color_rgb.png" style="width:40%">
+<img src="https://ysharma-dummy-chat-app.hf.space/file=/tmp/gradio/7dd7659cff2eab51f0f5336f378edfca01dd16fa/gemma_lockup_vertical_full-color_rgb.png" style="width:40%">
 <b>CodeGemma-7B-IT</b>
 """
 
 # Gradio block
 with gr.Blocks(fill_height=True) as demo:
-    gr.Markdown("# GEMMA-7b-IT")
+    gr.Markdown("# CODEGEMMA-7b-IT")
     #with gr.Tab('CodeGemma Chatbot'):
     gr.ChatInterface(codegemma, 
                    examples=[["Write a Python function to calculate the nth fibonacci number."]],