akhaliq HF Staff commited on
Commit
0ad5e40
·
verified ·
1 Parent(s): deec06c

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import spaces
5
 
6
  # Initialize the model pipeline
7
- model_id = "facebook/MobileLLM-1B"
8
  pipe = pipeline(
9
  "text-generation",
10
  model=model_id,
@@ -44,7 +44,7 @@ def respond(message, history):
44
  demo = gr.ChatInterface(
45
  fn=respond,
46
  title="MobileLLM Chat",
47
- description="Chat with Facebook's MobileLLM-1B model",
48
  examples=[
49
  "Write a Python function that returns the square of a number.",
50
  "Compute: 1-2+3-4+5- ... +99-100.",
 
4
  import spaces
5
 
6
  # Initialize the model pipeline
7
+ model_id = "facebook/MobileLLM-R1-950M"
8
  pipe = pipeline(
9
  "text-generation",
10
  model=model_id,
 
44
  demo = gr.ChatInterface(
45
  fn=respond,
46
  title="MobileLLM Chat",
47
+ description="Chat with Facebook's MobileLLM-R1-950M model",
48
  examples=[
49
  "Write a Python function that returns the square of a number.",
50
  "Compute: 1-2+3-4+5- ... +99-100.",