mrfakename commited on
Commit
91d32ae
1 Parent(s): 20be90d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -13,9 +13,9 @@ theme = gr.themes.Base(
13
  font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
14
  )
15
 
16
- tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/cosmo-1b", trust_remote_code=True)
17
  model = AutoModelForCausalLM.from_pretrained(
18
- "HuggingFaceTB/cosmo-1b",
19
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
20
  trust_remote_code=True,
21
  ).to(device)
@@ -33,17 +33,19 @@ def generate_text(text, temperature, maxLen):
33
  yield t
34
  with gr.Blocks(theme=theme) as demo:
35
  gr.Markdown("""
36
- # (Unofficial) Demo of Hugging Face's Cosmo 1B
37
 
38
  The model is suitable for commercial use and is licensed under the Apache license. I am not responsible for any outputs you generate. You are solely responsible for ensuring that your usage of the model complies with applicable laws and regulations.
39
 
40
- I am not affiliated with the authors of the model (Hugging Face).
 
 
41
 
42
  Note: for longer generations (>1024), keep clicking "Generate!" The demo is currently limited to 1024 demos per generation to ensure all users have access to this service. Please note that once you start generating, you cannot stop generating until the generation is done.
43
 
44
  By [mrfakename](https://twitter.com/realmrfakename).
45
 
46
- Duplicate this Space to skip the wait!
47
  """.strip())
48
  gr.DuplicateButton()
49
  text = gr.Textbox(label="Prompt", lines=10, interactive=True, placeholder="Write a detailed analogy between mathematics and a lighthouse.")
 
13
  font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
14
  )
15
 
16
+ tokenizer = AutoTokenizer.from_pretrained("yam-peleg/Experiment26-7B", trust_remote_code=True)
17
  model = AutoModelForCausalLM.from_pretrained(
18
+ "yam-peleg/Experiment26-7B",
19
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
20
  trust_remote_code=True,
21
  ).to(device)
 
33
  yield t
34
  with gr.Blocks(theme=theme) as demo:
35
  gr.Markdown("""
36
+ # (Unofficial) Demo of Experiment26-7B
37
 
38
  The model is suitable for commercial use and is licensed under the Apache license. I am not responsible for any outputs you generate. You are solely responsible for ensuring that your usage of the model complies with applicable laws and regulations.
39
 
40
+ I am not affiliated with the authors of the model.
41
+
42
+ **This model is not an instruct model but a base model.**
43
 
44
  Note: for longer generations (>1024), keep clicking "Generate!" The demo is currently limited to 1024 demos per generation to ensure all users have access to this service. Please note that once you start generating, you cannot stop generating until the generation is done.
45
 
46
  By [mrfakename](https://twitter.com/realmrfakename).
47
 
48
+ ## [Model card & download](https://huggingface.co/yam-peleg/Experiment26-7B)
49
  """.strip())
50
  gr.DuplicateButton()
51
  text = gr.Textbox(label="Prompt", lines=10, interactive=True, placeholder="Write a detailed analogy between mathematics and a lighthouse.")