SmallO commited on
Commit
fd2e2ee
·
1 Parent(s): 2e96f48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -1,11 +1,4 @@
1
  # -*- coding: utf-8 -*-
2
- """Untitled3.ipynb
3
-
4
- Automatically generated by Colaboratory.
5
-
6
- Original file is located at
7
- https://colab.research.google.com/drive/1eg0M7FRTlmONVU_5IaZH-EKADvdtODmT
8
- """
9
 
10
  #!pip install torch
11
 
@@ -20,7 +13,7 @@ import streamlit as st
20
  def pic_mo(prom):
21
  model_id = "runwayml/stable-diffusion-v1-5"
22
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # Change to float32
23
- # No need to specify .to("cuda") if running on CPU
24
 
25
  prompt = str(prom)
26
 
 
1
  # -*- coding: utf-8 -*-
 
 
 
 
 
 
 
2
 
3
  #!pip install torch
4
 
 
13
  def pic_mo(prom):
14
  model_id = "runwayml/stable-diffusion-v1-5"
15
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # Change to float32
16
+ # No need ("cuda") if running on CPU because dont't have money
17
 
18
  prompt = str(prom)
19